若有侵权,请通知删除,结果由 FaceRank AI 输出。git
由于隐私问题,训练图片集并不提供,稍微可能会放一些卡通图片。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!
提交 issue
QQ 群: 522785813
图片像素要提升
增长数据集
在临近的层次,用公用的图片:好比1-3;4-6;7-9 用类似或者相同图片。