在使用Pycharm训练数据集的时候，代码运行出现了问题

Question

import os
import random
import numpy as np
from skimage import io
from PIL import Image

root_dir = 'D:/ISIC 2018/'                # change it in your saved original data path
save_dir = 'D:/CA/data/ISIC2018_Task1_npy_all/'


if __name__ == '__main__':
    imgfile = os.path.join(root_dir, 'ISIC2018_Task1-2_Training_Input')
    labfile = os.path.join(root_dir, 'ISIC2018_Task1_Training_GroundTruth')
    filename = sorted([os.path.join(imgfile, x) for x in os.listdir(imgfile) if x.endswith('.jpg')])
    random.shuffle(filename)
    labname = [filename[x].replace('ISIC2018_Task1-2_Training_Input', 'ISIC2018_Task1_Training_GroundTruth'
                                   ).replace('.jpg', '_segmentation.png') for x in range(len(filename))]

    if not os.path.isdir(save_dir):
        os.makedirs(save_dir+'/image')
        os.makedirs(save_dir+'/label')

    for i in range(len(filename)):
        fname = filename[i].rsplit('/', maxsplit=1)[-1].split('.')[0]
        lname = labname[i].rsplit('/', maxsplit=1)[-1].split('.')[0]

        image = Image.open(filename[i])
        label = Image.open(labname[i])

        image = image.resize((342, 256))
        label = label.resize((342, 256))
        image = np.array(image)
        label = np.array(label)

        images_img_filename = os.path.join(save_dir, 'image', fname)
        labels_img_filename = os.path.join(save_dir, 'label', lname)
        np.save(images_img_filename, image)
        np.save(labels_img_filename, label)
    print('Successfully saved preprocessed data')

 

这里使用的是python3.8，下载的官网最新的pytorch，这里只是对论文作者的代码修改了文件路径，其他的没改。

Answer 1

image下面的ISIC2018_Task1-2_Training_Input这个文件夹不存在，你只创建到了image文件夹，但是你写入的时候是写道ISIC2018_Task1-2_Training_Input文件夹下面的，所以就报错了。估计你的label写入的时候也会有这问题，创建一下这层文件夹就好了