MUNIT的复现

Multimodal UnsupervisedImage-to-Image Translation的运行

下载预训练模型

https://drive.google.com/drive/folders/10IEa7gibOWmQQuJUIUOkh-CV4cm6k8__?usp=sharing 

并放入models文件夹

首先是运行，

我直接使用了CycleGAN的环境，但是出现了报错。

No module named 'torch.utils.serialization'

查询原因是因为pycharm版本太高，但是有其它的包可以解决这个问题，我就不去改低版本的pycharm了。解决方法参考

(55条消息) ModuleNotFoundError: No module named 'torch.utils.serialization' 的解决方案_悲恋花丶无心之人的博客-CSDN博客

需要将./utils.py文件中之前引用的from torch.utils.serialization import load_lua改为import torchfile。并将227行的load_lua改为torchfile.load。

def load_vgg16(model_dir):
    vgg = models.vgg16(pretrained=True)
    return vgg

需要安装torchfile以及yaml

pip install torchfile
pip install pyyaml

结果报错

    return yaml.load(stream)
TypeError: load() missing 1 required positional argument: 'Loader'

需要降级pyyaml 版本，将6.0降为5.4.1：

pip uninstall pyyaml
pip install pyyaml==5.4.1

顺利运行，可以测试了。线条生成包包，包包生成线条

python test.py --config configs/edges2handbags_folder.yaml --input inputs/edges2handbags_edge.jpg --output_folder outputs --checkpoint models/edges2handbags.pt --a2b 1

python test.py --config configs/edges2handbags_folder.yaml --input inputs/edges2handbags_handbag.jpg --output_folder outputs --checkpoint models/edges2handbags.pt --a2b 0

训练

使用 tensorboardX

pip install tensorboardX

./utils.py文件中 79 92行注释，因为我不希望RandomCrop，或者修改crop=True。