代码调试记录EnlightenGAN 一

代码及数据集链接

一、aligned_dataset.py 中

ModuleNotFoundError: No module named ‘data’

原因：python版本不同
论文源码要求python==3.5 我的是3.9
路径要求不同更改如下：

# from data.base_dataset import BaseDataset
# from data.image_folder import make_dataset
from base_dataset import BaseDataset
from image_folder import make_dataset

在“data”文件夹中的代码基本上都需要更改路径

---

二、combine_A_and_B.py 中

路径问题：

FileNotFoundError: [WinError 3] 系统找不到指定的路径。: ‘…/dataset/50kshoes_edges’

与上述一中的原因一致，需要将路径降低一个级

parser = argparse.ArgumentParser('create image pairs')
# parser.add_argument('--fold_A', dest='fold_A', help='input directory for image A', type=str, default='../dataset/50kshoes_edges')
# parser.add_argument('--fold_B', dest='fold_B', help='input directory for image B', type=str, default='../dataset/50kshoes_jpg')
# parser.add_argument('--fold_AB', dest='fold_AB', help='output directory', type=str, default='../dataset/test_AB')
parser.add_argument('--fold_A', dest='fold_A', help='input directory for image A', type=str, default='./50kshoes_edges')
parser.add_argument('--fold_B', dest='fold_B', help='input directory for image B', type=str, default='./50kshoes_jpg')
parser.add_argument('--fold_AB', dest='fold_AB', help='output directory', type=str, default='./test_AB')

自己创建文件夹并加入数据集

---

三、visualizer.py 中

ImportError: attempted relative import with no known parent package

原因同上
路径要求不同更改如下：

# from . import util
# from . import html
import util
import html
如果.后面带有文件夹 仅仅删除.即可 不需要文件夹

关于在py中路径问题：link

四、script.py 中

根据情况，修改了script.py中GPU卡号和batch_size:

  # --batchSize 32 \
  --batchSize 2 \

  # --gpu_ids 0,1,2 \
  --gpu_ids 0 \

---

五、train.py 中

ModuleNotFoundError: No module named ‘data’

原因：python版本不同
论文源码要求python==3.5 我的是3.9
路径要求不同更改如下：

pip install dominate

在“data”文件夹中的代码基本上都需要更改路径

---