1. 读取一个csv文件,预处理
def dataframe_creation():
df4 = pd.read_csv('**.csv')
df4['filename']='图片存放位置/'+df4['filename']
return df4
df = dataframe_creation()
这一步是为了去除标签中没有相应图像的标签,可选
listImg = os.listdir('图片存放位置')
string = '图片存放位置/'
list2 = list(map(lambda orig_string: string + orig_string , listImg))
indexify =[]
for i in df.index:
if df.iloc[i]['Left-Fundus'] in list2 and df.iloc[i]['Right-Fundus'] in list2:
continue
else:
indexify.append(i)
df = df.drop(indexify)
2.分文件夹存放
import pandas as pd
import os
import shutil
# 读取文件
file = open("E:/Classify/Ocular_disease_recoginition_data/8_image/name11.csv", "rb")
list = pd.read_csv(file)
# 创建文件夹,进行分类
for i in range(0, 7):
label_dir = os.path.join('A' +