前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >U-GAT-IT - 官方TensorFlow实施

U-GAT-IT - 官方TensorFlow实施

作者头像
代码医生工作室
发布2019-08-05 20:43:38
3.4K0
发布2019-08-05 20:43:38
举报
文章被收录于专栏:相约机器人

作者 | Junho Kim

来源 | Medium

编辑 | 代码医生团队

论文

https://arxiv.org/pdf/1907.10830.pdf

官方的Tensorflow代码

https://github.com/taki0112/UGATIT

官方的Pytorch代码

https://github.com/znxlwm/UGATIT-pytorch

抽象 提出了一种新的无监督图像到图像转换方法,它以端到端的方式结合了新的注意模块和新的可学习的归一化功能。注意模块引导模型基于辅助分类器获得的关注图来关注区分源域和目标域的更重要区域。与先前基于注意力的方法不同,这些方法无法处理域之间的几何变化,模型可以转换需要整体变化的图像和需要大变形的图像。此外,新的AdaLIN(自适应图层实例标准化)功能可帮助注意力引导模型根据数据集通过学习参数灵活地控制形状和纹理的变化量。

用法

├── dataset

└── YOUR_DATASET_NAME

├── trainA

├── xxx.jpg (name, format doesn't matter)

├── yyy.png

└── ...

├── trainB

├── zzz.jpg

├── www.png

└── ...

├── testA

├── aaa.jpg

├── bbb.png

└── ...

└── testB

├── ccc.jpg

├── ddd.png

└── ...

训练

代码语言:javascript
复制
python main.py --dataset selfie2anime

如果gpu的内存不足,请设置--light为True

测试

代码语言:javascript
复制
python main.py --dataset selfie2anime --phase test

Architecture

结果

消融研究

用户研究

对照

引文

如果发现此代码对研究有用,请引用论文:

代码语言:javascript
复制
@misc{kim2019ugatit,    title={U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation},    author={Junho Kim and Minjae Kim and Hyeonwoo Kang and Kwanghee Lee},    year={2019},    eprint={1907.10830},    archivePrefix={arXiv},    primaryClass={cs.CV}}
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-08-05,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 相约机器人 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档