目标检测数据增强：截取出bbox（VOC数据格式）

最新推荐文章于 2023-03-01 13:47:25 发布

winnerziqi

最新推荐文章于 2023-03-01 13:47:25 发布

阅读量572

点赞数 2

分类专栏： data augument 文章标签： python xml 大数据 c语言 .net

本文链接：https://blog.csdn.net/wty98wzq/article/details/114463843

版权

该代码段展示了如何利用Python的OpenCV和xml.etree.ElementTree库处理VOC数据集，遍历JPEGImages目录下的图片，读取对应的Annotations XML文件，解析出图像中的目标边界框，并截取每个目标区域保存为新的图片文件。

摘要由CSDN通过智能技术生成

截取出bbox（VOC数据格式）

import os
import cv2
import utils
try:
    import xml.etree.cElementTree as ET  # 解析xml的c语言版的模块
except ImportError:
    import xml.etree.ElementTree as ET

img_path = r"D:\Catalogue\master\generate_data\mitios_withxml\VOCdevkit2007\VOC2007\JPEGImages"  
xml_path = r"D:\Catalogue\master\generate_data\mitios_withxml\VOCdevkit2007\VOC2007\Annotations"  
img_names = os.listdir(img_path)
save_dir = r"D:\catalogue\master\generate_data\aug_paste"
utils.mkdir(save_dir) 

for imgname in img_names