我有一个pandas数据帧,其中第1列是我的文件路径,第2列是我的标签。是否可以根据标签的值将图像排序到新目录中。我使用3个标签,因此值为0、1、2
dataframe = pd.read_csv('\\extension.csv', header = True)
if dataframe.iat[0,1] == 0:
if not os.path.exists(dest_fold_0):
os.mkdir(dest_fold_0)
img.save(dest_fold_0 + dataframe.iat[1, 0], 'jpeg')
数据帧示例:
File_Extension,Labels
['\\test_0000_50x50.jpg'],0
['\\test_0001_50x50.jpg'],1
['\\test_0002_50x50.jpg'],0
['\\test_0003_50x50.jpg'],1
['\\test_0004_50x50.jpg'],2
['\\test_0005_50x50.jpg'],0
['\\test_0006_50x50.jpg'],1
目前没有回答
相关问题 更多 >
编程相关推荐