从voc xml文件中读取目标图像框的位置并把它从原图中截取出来

最新推荐文章于 2023-05-07 17:09:57 发布

gggoogle1020

最新推荐文章于 2023-05-07 17:09:57 发布

阅读量920

点赞数 3

本文链接：https://blog.csdn.net/qq_36249824/article/details/108428698

版权

voc数据集，xml格式

import cv2
import xml.etree.ElementTree as ET
import numpy as np

import xml.dom.minidom
import os
import argparse


def main():
    # JPG文件的地址
    img_path = './background/JPEGImages/'
    # XML文件的地址
    anno_path = './background/Annotations/'
    # 存结果的文件夹

    cut_path = './crops/'
    # 获取文件夹中的文件
    imagelist = os.listdir(img_path)
    # print(imagelist
    for image in imagelist:
        image_pre, ext = os.path.splitext(image)
        img_file = img_path + image
        img = cv2.imread(img_file)
        xml_file = anno_path + image_pre + '.xml'
        # DOMTree = xml.dom.minidom.parse(xml_file)
        # collection = DOMTree.documentElement
        # objects = collection.getElementsByTagName("object")

        tree = ET.parse(xml_file)
        root = tree.getroot()
        # if root.find('object') == None:
        #     return
        for obj in root.iter('object'):
            cls = obj.find('name').text
            xmlbox = obj.find('bndbox')
            b = [int(xmlbox.find('xmin').text), int(xmlbox.find('ymin').text), int(xmlbox.find('xmax').text),
                 int(xmlbox.find('ymax').text)]
            img_cut = img[b[1]:b[3], b[0]:b[2], :]
            path=os.path.join(cut_path,cls)
            # 目录是否存在,不存在则创建
            mkdirlambda = lambda x: os.makedirs(x) if not os.path.exists(x) else True
            mkdirlambda(path)
            cv2.imwrite(os.path.join(cut_path,cls,'cut_img_{}.jpg'.format(image_pre)), img_cut)

        # for object in objects:
        #     print("start")
        #     name=object.getElementsByTagName('name')[0]
        #     # obj.find('name').text
        #     print(name)
        #     print(type(name))
        #
        #     bndbox = object.getElementsByTagName('bndbox')[0]
        #     xmin = bndbox.getElementsByTagName('xmin')[0]
        #     xmin_data = xmin.childNodes[0].data
        #     ymin = bndbox.getElementsByTagName('ymin')[0]
        #     ymin_data = ymin.childNodes[0].data
        #     xmax = bndbox.getElementsByTagName('xmax')[0]
        #     xmax_data = xmax.childNodes[0].data
        #     ymax = bndbox.getElementsByTagName('ymax')[0]
        #     ymax_data = ymax.childNodes[0].data
        #     xmin = int(xmin_data)
        #     xmax = int(xmax_data)
        #     ymin = int(ymin_data)
        #     ymax = int(ymax_data)
        #     img_cut = img[ymin:ymax, xmin:xmax, :]
        #     cv2.imwrite(cut_path + 'cut_img_{}.jpg'.format(image_pre), img_cut)
            print("&&&&")


if __name__ == '__main__':
    main()