用Python写一段随机打乱voc数据集图片的代码,要求打乱图片的同时对应调整图片的xml文件,打乱后的图片和对应的xml文件存储在名为daluan的文件夹中
时间: 2023-05-28 15:02:25 浏览: 101
以下是一种可能的实现方式:
```python
import os
import random
import shutil
import xml.etree.ElementTree as ET
# 定义VOC数据集路径和随机打乱后的存储路径
voc_path = '/path/to/voc_dataset'
output_path = '/path/to/daluan'
# 创建目标文件夹
if not os.path.exists(output_path):
os.mkdir(output_path)
# 获取所有图片的路径
image_dir = os.path.join(voc_path, 'JPEGImages')
all_images = [f for f in os.listdir(image_dir) if os.path.isfile(os.path.join(image_dir, f))]
# 随机打乱图片路径和xml文件路径的对应关系
random.shuffle(all_images)
# 遍历所有图片和对应的xml文件,复制到目标文件夹中
for image_name in all_images:
# 复制图片文件
src_image_path = os.path.join(image_dir, image_name)
dst_image_path = os.path.join(output_path, image_name)
shutil.copyfile(src_image_path, dst_image_path)
# 复制对应的xml文件,并调整文件中的图片路径
xml_name = os.path.splitext(image_name)[0] + '.xml'
src_xml_path = os.path.join(voc_path, 'Annotations', xml_name)
dst_xml_path = os.path.join(output_path, xml_name)
shutil.copyfile(src_xml_path, dst_xml_path)
root = ET.parse(dst_xml_path).getroot()
for obj in root.findall('object'):
obj.find('bndbox/xmin').text = str(int(obj.find('bndbox/xmin').text))
obj.find('bndbox/xmax').text = str(int(obj.find('bndbox/xmax').text))
obj.find('bndbox/ymin').text = str(int(obj.find('bndbox/ymin').text))
obj.find('bndbox/ymax').text = str(int(obj.find('bndbox/ymax').text))
root.find('filename').text = image_name
root.find('path').text = dst_image_path
ET.ElementTree(root).write(dst_xml_path)
print('Done!')
```
这段代码首先定义了VOC数据集路径和随机打乱后的存储路径,在代码中分别赋值给`voc_path`和`output_path`变量。然后,创建了目标文件夹`output_path`,如果不存在的话。
接着,获取了所有图片的路径(默认存储在`JPEGImages`子文件夹中),使用`random.shuffle()`函数随机打乱了图片文件名列表,从而得到了图片路径和对应的xml文件路径的随机打乱的对应关系。
最后,遍历所有图片和对应的xml文件,复制到目标文件夹中。对于每个图片文件,直接使用`shutil.copyfile()`函数复制到对应的位置;对于对应的xml文件,先使用`shutil.copyfile()`函数复制到对应的位置,然后读取该xml文件的内容,使用`xml.etree.ElementTree`模块解析xml文件,调整其中的图片路径和标注框坐标,保存修改后的xml文件。
运行以上代码后,所有随机打乱后的图片和xml文件就会存储在新建的`daluan`文件夹中。需要注意的是,这段代码假定VOC标注文件中的图片路径字段是`path`,以此来在xml文件中修改打乱后的图片路径。如果你的VOC标注文件中的图片路径字段不同,请修改代码中的相应字段即可。
阅读全文