「深度學習測顏值」基於TensorFlow的開源項目FaceRank

1 新智元推薦

我們常看到用機器學習識別字體，自動駕駛等項目，今天給大家推薦一個有趣的項目 FaceRank，這是個開源項目，它基於 TensorFlow CNN 模型，提供了一些圖片處理的工具集，後續還會提供訓練好的模型。

從此以後，讓它來幫你尋找高顏值的小電影，幫你篩選附近高顏值的妹子（漢子），讓它幫你給學校或者公司帥哥美女做個排行榜，讓它給明星打分並且你可以自豪的說「一切都是人工智慧的選擇」。。。

以下是機器給蒼老師的打分。

（機器給蒼老師打了 7 分，這已經是很高的分數了，果然德藝雙磬）

數據集

• 130 張 128*128 張網路圖片，圖片名：1-3.jpg 表示 分值為 1 的第 3 張圖。 你可以把符合這個格式的圖片放在 resize_images 來訓練模型。

• find_faces_in_picture.py

find_and_save_face 基於 face_recognition 從圖片中找到人臉的坐標，並保存為新圖片。

• 然後再用 resize 統一為 128×128 大小，為模型訓練做準備。

模型

人臉打分基於 TensorFlow 的 CNN 模型 代碼參考 : TensorFlow-Examples

卷積神經網路部分代碼，網路結構說明：

• 卷積層

• 池化層

• 卷積層

• 池化層

• 全鏈接層

運行

安裝好 TensorFlow 之後，直接運行 train_model.py .

• 訓練模型

• 保存模型到 model 文件夾

測試

運行完 train_model.py 之後,直接運行 run_model.py 來測試.

下載

訓練好的模型下載網址： （文件較大，正在上傳） http://www.tensorflownews.com/

模型效果

• 訓練過程 你可以看訓練過程：Train_Result.md ,這裡有損失函數和準確率變化過程。

• 測試結果 結果並不非常好，但是增加數據集之後有所改善。

支持

• 提交 issue

• Github: https://github.com/fendouai/FaceRank

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！