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

AI 科技评论按:本文作者灰灰,本文原载于作者的知乎专栏。授权转载。

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

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

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

苍老师镇楼:

隐私

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

数据集

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

测试

运行完 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

Github: https://https://github.com/fendouai/FaceRank

原文发布于微信公众号 - AI科技评论(aitechtalk)

原文发表时间:2017-08-03

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏AI科技评论

开发 | 无需编程,仅用摄像头,Google最新项目让你3分钟学会机器学习

得益于各种套件,今天的机器学习的门槛已经越来越低。但Google显然并不满足于此,其最新推出了Teachable Machine项目,让用户无需编程就可以用手机...

2758
来自专栏量子位

深度学习动手入门:GitHub上四个超棒的TensorFlow开源项目

问耕 编译自 Source Dexter 量子位 出品 | 公众号 QbitAI 作者简介:akshay pai,数据科学工程师,热爱研究机器学习问题。Sour...

4729
来自专栏CSDN技术头条

AI 可能真的要代替插画师了……

事先声明,这篇文章的标题绝不是在耸人听闻。事情的起因是前段时间在朋友圈看到同学在转发一篇论文,名字叫《Create Anime Characters with ...

1858
来自专栏人工智能头条

资源 | 深度学习图像标注工具汇总

1141
来自专栏磐创AI技术团队的专栏

FaceRank-人脸打分基于 TensorFlow 的 CNN 模型,这个妹子颜值几分?

FaceRank-人脸打分基于 TensorFlow 的 CNN 模型 机器学习是不是很无聊,用来用去都是识别字体。能不能帮我找到颜值高的妹子,顺便提高一下姿势...

3794
来自专栏IT派

资源 | 深度学习图像标注工具汇总

对于监督学习算法而言,数据决定了任务的上限,而算法只是在不断逼近这个上限。世界上最遥远的距离就是我们用同一个模型,但是却有不同的任务。但是数据标注是个耗时耗力的...

882
来自专栏AI科技大本营的专栏

机器学习不神秘!手把手教你用R语言打造文本分类器

简单安装几个R软件包,你就直接在自己电脑上打造出一个文本分类器,用进行机器来评估人类写作。 本文是一篇极简的上手教程,它想用清晰明了的步骤告诉读者,机器学习不...

34416
来自专栏AI科技评论

学界 | 传播动态学的主动监控:一种组稀疏贝叶斯学习方法

AI 科技评论按:本文作者吉林大学博士生裴红斌,本文为对他发表在 AAAI 2018 论文的独家解读稿件,未经许可不得转载。 Group Sparse Bay...

2886
来自专栏机器之心

库、教程、论文实现,这是一份超全的PyTorch资源列表(Github 2.2K星)

项目地址:https://github.com/bharathgs/Awesome-pytorch-list

1034
来自专栏灯塔大数据

塔趣 | 用深度学习来评判颜值,基于TensorFlow的开源项目FaceRank

导读:用深度学习来评判颜值,已开源。好友 @小灰灰 大大的「颜值评分 FaceRank」,这是基于 TensorFlow 的 CNN 模型,美不美机器说了算。...

39711

扫码关注云+社区