我有一个数据集,其中一个文件夹包含图像,另一个文件夹包含相应的文本文件。每个文本文件包含相应类的标签。
Images folder
image_0000.jpeg
image_0001.jpeg
Label folder
image_0000.txt
image_0001.txt
标签文本文件包含值0或1或2。
我想保存图像对应的标签0在另一个文件夹。类似的其余标签1,2
正如你在下面的图片中看到的。
Dataset Description
def read_image_list(image_list_file):
f = open(image_list_file, 'r')
filenames = []
for line in f:
filename, label = line[:-1].split(' ')
filenames.append(filename)
return filenames
2条答案
按热度按时间bvn4nwqk1#
1.导入必要的库,如os、shutil和PIL。
1.定义读取和保存图像的源目录和目标目录。
1.循环访问源目录中的每个文件,并使用PIL库检查它是否是图像文件。
1.如果文件是图像,则通过将扩展名从“.jpg”更改为“.txt”来读取其对应的文本文件。
1.从文本文件中提取标签,并在目标目录中使用该标签创建新文件夹。
1.使用shutil将图像文件复制到新创建的文件夹中。
1.对源目录中的每个图像文件重复上述步骤。
下面是实现上述步骤的代码:
5us2dqdw2#
只要改变前三个目录变量就会让代码像魔咒一样工作,Prem J给出的答案也很酷。