AdaIN 代码复现

1、环境配置

conda create -n adain python=3.9
conda activate adain

pip install -r requirements.txt

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

 

2、train

模型下载：Release v0.0.0 · naoto0804/pytorch-AdaIN

下载 decoder.pth / vgg_normalized.pth 到 models/ 文件夹

修改 train.py 文件中的 数据集路径，并且删除后面的 required = True

训练：

python train.py

3、test

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