眼底OCT图像疾病8分类数据集预处理

[Auroa-smart]

你好，请问您给的代码里面并没有mask数据集，我从kaggle上下载的数据集都是jpg图片，那您对jpg数据集的预处理代码方便给一下吗？或者解答一下是怎么处理得到mask的？

[junqiangchen]

分类的标签需要自己进行映射，也可以根据目录的顺序赋予对应数值标签即可

[junqiangchen]

你好，请问您给的代码里面并没有mask数据集，我从kaggle上下载的数据集都是jpg图片，那您对jpg数据集的预处理代码方便给一下吗？或者解答一下是怎么处理得到mask的？

def save_file2csvclassify(file_dir, file_name):
"""
save file path to csv
:param file_dir:preprocess data path
:param file_name:output csv name
:return:
"""
out = open(file_name, 'w')
labels = file_name_path(file_dir, True, False)
for index in range(len(labels)):
out_file_image_paths = file_dir + "/" + labels[index]
imagepaths = file_name_path(out_file_image_paths, False, True)
for file in range(len(imagepaths)):
out.writelines(str(index) + "," + out_file_image_paths + '/' + imagepaths[file] + "\n")
out.close()