训练一个AI给颜值打分,公平公正!

训练一个AI给颜值打分,公平公正!

时间:2020-01-10 13:26 作者:admin 点击:
阅读模式 16:04 导语:到底是谁比较好看,现在有公平的裁判了~ 雷锋网 (公众号:雷锋网) 按:本文作者 灰灰 ,本文原载于作者的 知乎专栏 。雷锋网获得授权转载。

机器学习是不是很无聊,用来用去都是识别字体。能不能帮我找到颜值高的妹子,顺便提高一下姿势水平。

FaceRank 基于 TensorFlow CNN 模型,提供了一些图片处理的工具集,后续还会提供训练好的模型。给 FaceRank 一个妹子,他给你个分数。

从此以后筛选简历,先把头像颜值低的去掉;自动寻找女主颜值高的小电影;自动关注美女;自动排除负分滚粗的相亲对象。从此以后升职加薪,迎娶白富美,走上人生巅峰。

苍老师镇楼:

amp;lt;img src= amp;quot;https://static.leiphone.com/uploads/new/article/pic/201707/b77a84c41f4d8205362b970b26b9736b.jpg amp;quot; data-rawwidth= amp;quot;766 amp;quot; data-rawheight= amp;quot;570 amp;quot; class= amp;quot;origin_image zh-lightbox-thumb amp;quot; width= amp;quot;766 amp;quot; data-original= amp;quot;https://pic1.zhimg.com/v2-311b070ef3f1443307d9a0bb7d1af604_r.jpg amp;quot; _src= amp;quot;https://static.leiphone.com/uploads/new/article/pic/201707/b77a84c41f4d8205362b970b26b9736b.jpg amp;quot;/ amp;gt;

隐私

因为隐私问题,训练图片集并不提供,但是提供了人脸抽取,图片大小归一化工具,稍微可能会放一些卡通图片。

数据集

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 文件夹