专栏首页IT派Github 推荐项目 | 用 TensorFlow 简单地实现 StarGAN

Github 推荐项目 | 用 TensorFlow 简单地实现 StarGAN

本库用 TensorFlow 简单的实现了 StarGAN。

StarGAN 是一种新颖且可扩展的方法,可以仅使用一个模型来执行多个域的图像到图像的转换。StarGAN 这样一个统一的模型体系架构让开发者可以同时训练单个网络中具有不同域的多个数据集,这导致StarGAN的图像转化结果比现有模型质量更高,并具有将输入图像灵活转化成任何期望目标域的新颖能力。

Github:

https://github.com/taki0112/StarGAN-Tensorflow

依赖

  • Tensorflow 1.8
  • Python 3.6

用法

下载数据集:

> python download.py celebA

├── dataset └── celebA ├── train ├── 000001.jpg ├── 000002.jpg └── ... ├── test (It is not celebA) ├── a.jpg (The test image that you wanted) ├── b.png └── ... ├── list_attr_celeba.txt (For attribute information)

训练:

  • python main.py --phase train

测试:

  • python main.py - 阶段测试
  • 同时运行 celebA 测试图像和您想要的图像

预训练模型:

  • Download checkpoint for 128x128 https://drive.google.com/open?id=1ezwtU1O_rxgNXgJaHcAynVX8KjMt0Ua-

总结

结果(128x128, wgan-gp):

女人

男人

本文分享自微信公众号 - IT派(transfer_3255716726)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-06-24

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 教你用Python来玩微信跳一跳实现物理外挂

    1、下载程序:打开下面的链接,点右侧 clone or download,再点 download zip;

    IT派
  • 推荐|数据科学家需要了解的5大聚类算法

    IT派 - {技术青年圈} 持续关注互联网、大数据、人工智能领域 聚类是一种涉及数据点分组的机器学习技术。给定一个数据点集,则可利用聚类算法将每个数据点分类...

    IT派
  • 机器学习中导数最优化方法(基础篇)

    1. 前言 熟悉机器学习的童鞋都知道,优化方法是其中一个非常重要的话题,最常见的情形就是利用目标函数的导数通过多次迭代来求解无约束最优化问题。实现简单,codi...

    IT派
  • Github 推荐项目 | 用 TensorFlow 简单地实现 StarGAN

    StarGAN 是一种新颖且可扩展的方法,可以仅使用一个模型来执行多个域的图像到图像的转换。StarGAN 这样一个统一的模型体系架构让开发者可以同时训练单个网...

    AI研习社
  • 玩转FPGA边缘视觉——4k视频图像抓取

    随着现代图像及视频处理技术的不断发展,人们对图像处理提出了新的要求,最近几年,图像的分辨率和扫描频率都有了较大范围的提升,1080P分辨率的视频已经非常流行,2...

    用户7124040
  • ns3生成.pcap文件

    相关函数有EnablePcap()和EnalePcapAll(),     支持第一个函数的类有ns3::YansWifiPhyHelper  PointToP...

    py3study
  • TensorFlow On Spark 开源项目分析

    原文:http://sparkdata.org/?p=423&utm_source=tuicool&utm_medium=referral 作者:京东大数据技术...

    小莹莹
  • tensorflow:AToolDeveloperGuideToTFModelFIles

    Tensorflow Model Files 最近闲来无聊,想深入理解一下tensorlfow,也不知从何下手,突然间发现了官方文档的Extend模块下还有这个...

    ke1th
  • PHP开发微信公共平台(验证token)ZendFramework

    define('TOKEN', '3FC50DEAED1083F162BB3D36FF053709'); //这个是TOKEN,...

    苦咖啡
  • 中国人群肺癌队列的多组学探索(逆向收费读文献2020-13)

    多组学通常就是全外显子和转录组而已,这个规律早就提现在了各个国家地区的队列之中(之前我们分享过韩国人的乳腺癌队列和台湾人的口腔癌队列),本研究也不例外,发表在:...

    生信技能树

扫码关注云+社区

领取腾讯云代金券