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

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 用類似或者相同圖片。