专栏名称: 新智元
智能+中国主平台,致力于推动中国从互联网+迈向智能+新纪元。重点关注人工智能、机器人等前沿领域发展,关注人机融合、人工智能和机器人革命对人类社会与文明进化的影响,领航中国新智能时代。
目录
相关文章推荐
51好读  ›  专栏  ›  新智元

【深度学习测颜值】基于TensorFlow的开源项目FaceRank

新智元  · 公众号  · AI  · 2017-08-01 12:58

正文

请到「今天看啥」查看全文


1 新智元推荐


来源:知乎

作者:园长


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



我们常看到用机器学习识别字体,自动驾驶等项目,今天给大家推荐一个有趣的项目 FaceRank,这是个开源项目,它基于 TensorFlow CNN 模型,提供了一些图片处理的工具集,后续还会提供训练好的模型。


从此以后,让它来帮你寻找高颜值的小电影,帮你筛选附近高颜值的妹子(汉子),让它帮你给学校或者公司帅哥美女做个排行榜,让它给明星打分并且你可以自豪的说「一切都是人工智能的选择」。。。


以下是机器给苍老师的打分。


(机器给苍老师打了 7 分,这已经是很高的分数了,果然德艺双磬)


数据集


  • 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 ,这里有损失函数和准确率变化过程。

  • 测试结果 结果并不非常好,但是增加数据集之后有所改善。


支持
  • 提交 issue

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









请到「今天看啥」查看全文


推荐文章
经典短篇阅读小组  ·  在美国地铁里的一个实验
8 年前
经典短篇阅读小组  ·  在美国地铁里的一个实验
8 年前
肥肥猫的小酒馆  ·  春节放假,你准备好应付亲戚的问题了吗?
8 年前