FaceRank-人臉打分基於 TensorFlow 的 CNN 模型,這個妹子顏值幾分? FaceRank 告訴你!

FaceRank-人臉打分基於 TensorFlow 的 CNN 模型

結果圖片

若有侵權,請通知刪除,結果由 FaceRank AI 輸出。
Result Picgit

隱私

由於隱私問題,訓練圖片集並不提供,稍微可能會放一些卡通圖片。github

數據集

  • 130張 128*128 張網絡圖片,圖片名: 1-3.jpg 表示 分值爲3 的第3 張圖。
    你能夠把符合這個格式的圖片放在 resize_images 來訓練模型。網絡

模型

人臉打分基於 TensorFlow 的 CNN 模型 代碼參考 : https://github.com/aymericdam...函數

運行

安裝好 TensorFlow 以後,直接運行 train_model.py.測試

  • 訓練模型spa

  • 保存模型到 model 文件夾code

測試

運行完 train_model.py 以後,直接運行 run_model.py 來測試.圖片

下載

訓練好的模型能夠在如下網址下載:
http://www.tensorflownews.com/rem

模型效果

  • 訓練過程
    你能夠看訓練過程: Train_Result.md ,這裏有損失函數和準確率變化過程。get

  • 測試結果
    結果並不很是好,可是增長數據集以後有所改善。

(?, 128, 128, 24)
(?, 64, 64, 24)
(?, 64, 64, 96)
(?, 32, 32, 96)

['1-1.jpg', '1-2.jpg', '10-1.jpg', '10-2.jpg', '2-1.jpg', '2-2.jpg', '3-1.jpg', '3-2.jpg', '4-1.jpg', '4-2.jpg', '5-1.jpg', '5-2.jpg', '6-1.jpg', '6-2.jpg', '7-1.jpg', '7-2.jpg', '8-1.jpg', '8-2.jpg', '9-1.jpg', '9-2.jpg']
20
(10, 128, 128, 3)
[3 2 8 6 5 8 0 4 7 7]
(10, 128, 128, 3)
[2 6 6 6 5 8 7 8 7 5]
Test Finished!

支持

後續計劃

  • 圖片像素要提升

  • 增長數據集

  • 在臨近的層次,用公用的圖片:好比1-3;4-6;7-9 用類似或者相同圖片。

相關文章
相關標籤/搜索