根据提供的csv生成文件夹数据

像pascal voc2012这样提供的valid.txt或者其他valid.csv,现在根据csv把数据集抽离出来,

#-*- coding : utf-8-*-
import os, sys
import shutil
import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import roc_curve, auc
import pandas as pd                      #导入pandas库

# coding:unicode_escape
excel_file = r'C:/Users/Desktop/valid.csv'    #导入csv数据
x = pd.read_csv(excel_file, encoding="gbk")
x = x.iloc[0:,0]  #x的类型为"pandas.core.series.Series"
num1 = x.size
print(x)
print(type(x))
a = []
b = []
tar_path = "/data2/liguanhua/Datasets/valid"
for i in range(num1):
    img_name = x[i].split('/')[-1]
    tar_imgpath = tar_path + "/" + img_name
    shutil.copy(x[i], tar_imgpath)
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容