FaceRank-人臉打分基於 TensorFlow 的 CNN 模型
隱私
因為隱私問題，訓練圖片集並不提供，稍微可能會放一些卡通圖片。
數據集
- 130張 128*128 張網絡圖片，圖片名： 1-3.jpg 表示 分值為3 的第3 張圖。
你可以把符合這個格式的圖片放在 resize_images 來訓練模型。
模型
人臉打分基於 TensorFlow 的 CNN 模型 代碼參考 :
https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/3_NeuralNetworks/convolutional_network.py
運行
安裝好 TensorFlow 之后，直接運行 train_model.py.
- 訓練模型
- 保存模型到 model 文件夾
測試
運行完 train_model.py 之后,直接運行 run_model.py 來測試.
下載
訓練好的模型可以在以下網址下載：
http://www.tensorflownews.com/
模型效果
- 訓練過程
你可以看訓練過程： Train_Result.md ,這里有損失函數和准確率變化過程。 - 測試結果
結果並不非常好，但是增加數據集之后有所改善。
(?, 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!
支持
- 提交 issue
- QQ 群: 522785813
后續計划
- 圖片像素要提高
- 增加數據集
- 在臨近的層次，用公用的圖片：比如1-3；4-6；7-9 用相似或者相同圖片。