GAN 非平稳纹理合成

该库是论文「Non-stationary texture synthesis using adversarial expansions.」的官方代码。

论文链接:

http://vcc.szu.edu.cn/research/2018/TexSyn

Github 链接:

https://github.com/jessemelpolio/non-stationary_texture_syn

如果你需要使用此代码做研究,请引用以下内容:

@article,

author = ,

journal = ,

volume = {37},

number = {4},

pages = {},

year = {2018},

}

需求

此代码在 Ubuntu 14.04 和 16.04 下进行了测试。 整个项目可以在以下环境中运作良好:

python-2.7

正确指定了 CUDA 的 pytorch-0.3.0

CUDA 8.0

python-2.7 下的其他软件包

准备工作

首先运行,download_pretrained_models.sh 来创建一个新的 Models 文件夹,然后将在 ImageNet 上预先训练过的 VGG19 模型下载到此文件夹,预训练的 VGG19 模型用于计算 style loss。

数据

源纹理图片没有格式的限制,建议数据文件夹结构使用 datasets 文件夹中子文件夹中的结构。

+--half

|

| +--sunflower

|

| +--train

|

| +--sunflower.jpg

|

| +--test

|

| +--sunflower.jpg

|

| +--brick

|

| +--train

|

| +--brick.jpg

|

| +--test

|

| +--brick.jpg

|

...

该库的结构

主文件夹,train.py 用于训练论文中描述的模型,test.py 用于测试原始图像(结果是输入大小的两倍), test_recurrent.py 用于极端扩展,nn-vis.py 用于可视化发生器的内部层,论文中显示的 residual blocks 可视化通过 cnn-vis.py 生成。

在 data 文件夹数据中,文 件custom_dataset_data_loader 指定了五种数据集模式:aligned,unaligned,single 和 half_crop,一般来讲,用 single 来测试,用 half_crop 来训练。

在 models 文件夹中,两个文件非常重要:models.py 和 networks.py,请在使用前仔细检查它。half_gan_style.py 是论文中使用的主要模型。一些实用程序在 vgg.py 中实现。

在 options 文件夹中,所有超参数都在这里定义。转到此文件夹可查看每个超参数的含义。

script 文件夹里包含了用于测试和训练的脚本。要训练或测试模型,请使用诸如 sh scripts / train_half_style.sh 之类的命令。

util 文件夹存放了一些用于产生柏林噪声的脚本(perlin2d.py),生成随机区块(random_tile.py),这对复制我们论文的结果非常有用。 其他一些有用的脚本也包括在内。

从Python入门-如何成为AI工程师

BAT资深算法工程师独家研发课程

最贴近生活与工作的好玩实操项目

班级管理助学搭配专业的助教答疑

学以致用拿offer,学完即推荐就业

用 Keras 搭建 GAN:图像去模糊中的应用(附代码)

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180607A084ZF00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券