字体风格迁移MC-GAN运行记录

一，代码

代码

二，环境配置

Linux or macOS
Python 2.7#python3不行
CPU or NVIDIA GPU + CUDA CuDNN

pip install visdom
pip install dominate
pip install scikit-image

三，下载文件

mkdir FontTransfer
cd FontTransfer
git clone https://github.com/azadis/MC-GAN
cd MC-GAN

下载文件，在MC-GAN文件夹中打开终端，输入如下指令
文件1

./datasets/download_font_dataset.sh Capitals64

文件2

./datasets/download_font_dataset.sh public_web_fonts

文件3

./datasets/download_font_dataset.sh Capitals_colorGrad64

会在MC-GAN文件夹的同级目录生成一个datasets文件夹

四，训练

同样在MC-GAN文件夹下运行

./scripts/train_cGAN.sh Capitals64

问题1：python2下没有torch。
解决方法：卸载python3的torch，从装python2的pytorch
问题2：TypeError: cuda() got an unexpected keyword argument ‘device_id’
解决方法：找到代码中所有的device_id，替换为device
问题3：ImportError: No module named skvideo.io
解决方法：pip install scikit-video
问题4：pytorch1.4之后不再支持python2了
问题5：pytorch安装太慢
解决方法：pip install torch1.4.0 torchvision0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple 即后面加上清华镜像。
问题6：MC-GAN/models/cGAN_model.py", line 250出现IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
解决方法：loss.data[0]改为loss.item()
问题7：RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
解决方法：在train.py加入
import torch
torch.cuda.set_device(0)