我有一个pandas数据框,其中第1列是我的file_path,第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')
DataFrame示例:
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