问题修复:PaddleCV/gan/data_reader.py
Created by: shaoshenchen
我在开始训练CycleGAN模型时项目报错,错误信息为:
FileNotFoundError: [Errno 2] No such file or directory: 'data/vangogh2photo/trainB/2015-05-18'
后来发现在 data_reader.py 中对文件名分割错误导致的。
在 trainB
文件中,图片的格式为 日期+时间 ,例如 2016-01-20 08:00:45.jpg,但是通过 line = line.strip('\n\r\t').split(' ')
代码分割后,日期和时间被作为列表中两个独立的元素,在后来的 img = Image.open(os.path.join(self.image_dir, file)).convert(
代码中直接相连,导致无法找到图片。
解决方案:
将line = line.strip('\n\r\t').split(' ') 替换成line = line.strip('\n\r\t').split('\t')