AdaIN风格迁移实验

参考源码

环境配置

pytorch环境

直接参考pytorch官网，输入命令行：

# 由于这个安装很容易因为网络不好中断，我把torch，torchvision分开装。多试几次就成功了。
pip3 install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

插播

安装太慢了。
换用镜像，改pip镜像源：但我这样操作了，其实没啥用。因为命令中-f https://download.pytorch.org/whl/torch_stable.html导致还是去官网上找。使用镜像应该有对应的命令。

mkdir ~/.pip 
vim ~/.pip/pip.conf
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host = mirrors.aliyun.com

# 附上国内镜像源
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣 http://pypi.douban.com/simple
Python官方 https://pypi.python.org/simple/
v2ex http://pypi.v2ex.com/simple/
中国科学院 http://pypi.mirrors.opencas.cn/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

其他依赖

官方给的requirements.txt如下：

numpy==1.17.2
Pillow==6.1.0
pkg-resources==0.0.0
protobuf==3.9.1
six==1.12.0
tensorboardX==1.8
torch==1.2.0
torchvision==0.4.0
tqdm==4.35.0

因为我想把pytorch环境装得新一点，所以之前一步单独装。requirements.txt删除部分内容，如下：

numpy==1.17.2
Pillow==6.1.0
pkg-resources==0.0.0
protobuf==3.9.1
six==1.12.0
tensorboardX==1.8
tqdm==4.35.0

使用这个文件方法：

cd /home/user/adain/pytorch-AdaIN-master
pip3 install -r requirements.txt

中途可能会有pkg-resources==0.0.0找不到提示，那把这行删除就行，我是这么做的。然后环境就装好了。

测试

把一张content下图片转换成style下图片的风格

cd /home/user/adain/pytorch-AdaIN-master
python3 test.py --content input/content/cornell.jpg --style input/style/woman_with_hat_matisse.jpg

把content目录下图片，两两对应转换成style下图片的风格

python3 test.py --content_dir input/content --style_dir input/style

多种风格混合

python3 test.py --content input/content/avril.jpg --style input/style/picasso_self_portrait.jpg,input/style/impronte_d_artista.jpg,input/style/trial.jpg,input/style/antimonocromatismo.jpg --style_interpolation_weights 1,1,1,1 --content_size 512 --style_size 512 --crop
# 选项意思如下：
--content_size: New (minimum) size for the content image. Keeping the original size if set to 0.
--style_size: New (minimum) size for the content image. Keeping the original size if set to 0.
--alpha: Adjust the degree of stylization. It should be a value between 0.0 and 1.0 (default).
--preserve_color: Preserve the color of the content image.

训练

python3 train.py --content_dir ./input/content/ --style_dir ./input/style/ # python3 train.py --content_dir  --style_dir