已有一个名为JPEG包含100车辆的图片文件集(格式jpg),有一个名为GJSB包含100个txt格式的文件(里面的每一个txt信息是BUS,SVU,CAR中的其中一个)如何将这个GJSB文件里的信息一一对应到JPEG的每一张图片里)
时间: 2024-02-24 22:58:47 浏览: 129
已经标记好的深度学习车辆类型数据集,xml和txt两种格式,1543张图片,共三个类别
5星 · 资源好评率100%
首先,你可以将GJSB文件夹中的所有txt文件读入一个列表中。然后,你可以使用Python中的Pillow库来打开JPEG文件,并使用文件名中的数字来标识每个文件。在for循环中,你可以使用相同的索引来读取与之相关的txt文件,然后将txt文件中的内容添加到JPEG文件的元数据中。下面是Python代码示例:
```
import os
from PIL import Image
# 读取GJSB文件夹中的所有txt文件
txt_files = []
for filename in os.listdir('GJSB'):
if filename.endswith('.txt'):
txt_files.append(filename)
# 处理JPEG文件
for i in range(1, 101): # 假设JPEG文件编号从1到100
image_filename = f'JPEG/{i}.jpg'
txt_filename = f'GJSB/{txt_files[i-1]}'
# 打开JPEG文件并添加元数据
with Image.open(image_filename) as img:
with open(txt_filename) as txt:
txt_content = txt.read().strip()
img.info['vehicle_type'] = txt_content
# 保存修改后的JPEG文件
img.save(image_filename)
```
这段代码将遍历JPEG文件夹中的所有JPEG文件,并使用与之相关的txt文件来添加元数据。在这里,我们假设JPEG文件编号从1到100,并且文件名格式为`{数字}.jpg`。注意,这段代码只是一个示例,你需要根据自己的实际情况进行修改。
阅读全文