对DW-GAN进行训练时,自己编写了train.py代码,执行时出现以下问题:

FileNotFoundError: [Errno 2] No such file or directory: './train/hazy/hazy'

 这个问题是文件路径有误导致的,可通过如下语句解决:

def __init__(self, train_dir):
    self.transform = transforms.Compose([transforms.ToTensor()])
    self.list_train=[]

    for i in os.listdir(train_dir+'/hazy'):
        self.list_train.append(i)

    #'./train/'
    self.root_hazy=os.path.join(train_dir, 'hazy/')
    self.root_clean =os.path.join(train_dir, 'clean/')
    self.file_len = len(self.list_train)

在后面调用Image.open时会定位到train/hazy文件夹

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐