CSV多标签数据预处理+加上csv多标签转文件夹+ ISIC2018数据集分类预处理

最新推荐文章于 2024-06-21 09:48:57 发布

弗吉尼亚有个约会

最新推荐文章于 2024-06-21 09:48:57 发布

阅读量3.3k

点赞数 5

文章标签： python 深度学习

本文链接：https://blog.csdn.net/qq_44504069/article/details/127940330

版权

在下小白,深感自己的代码能力太差劲的,真要是大佬,那不是分分钟写出代码;

我这个是准备跑分类网络,有些模型要求分类的数据集必须是文件夹那种类型已经分好了,我找到的是官网的数据集,没有进行数据预处理;

在我处理数据的时候,在网上搜索关于csv针对多标签处理的时候,发现我没找到好用的代码;

在综合好几个博主的代码放在这里,参考的太多了,没时间就不放了;

实测也可以运行

import pandas as pd
import os
import shutil  

# 读取表格文件+填写你的csv文件的位置
f = open(r"D:\PythonProject\dataset\ISIC2018_Task3_Training_GroundTruth.csv", "rb") 
list = pd.read_csv(f)

# 进行分类----填写你要分类文件夹的标签,有多少就写多少
for i in ['NV','DF','AKIEC','VASC','BKL','MEL','BCC']:
    if not os.path.exists(i):
        os.mkdir(i)
    listnew = list[list["type"] == i]#type是你csv文件里面的你要处理的那一列的列名称
    l = listnew["image"].tolist()#image这里是你的处理文件的名字的列名称
    j = str(i)
    for each in l:
        #这里是你数据文件放置的位置
        shutil.copy('D:\\PythonProject\\dataset\\ISIC2018_Task3_Training_Input\\images\\' + each, j)

在这里放上我的数据csv文件的处理的结果