首先,介绍一下要被分类存储的数据集格式
Base文件夹中的图片有两个标签,代表两个指标,图片存储在Base文件夹中,指标一和二的标签都在Annotation_Base.xls中,list.txt存放的是图片的路径和该图片对应的指标一;list1.txt存放的是图片的路径和该图片对应的指标二
分好的图片数据集格式长这个样子
import numpy as np
import os
import shutil
#****************将图片数据按照标签存储至文件夹***************
root='E:/Python/project/DIY/MESSIDOR/Risk of macular edema' #不同指标需要建立不同的存储文件夹
#创建文件夹
for i in range(4):
os.mkdir(root+'/'+str(i