通过xml文件给图像加目标框的python模块

最新推荐文章于 2024-07-19 04:06:16 发布

JamesWH

最新推荐文章于 2024-07-19 04:06:16 发布

阅读量1.4k

点赞数

分类专栏： python 文章标签： python xml

本文链接：https://blog.csdn.net/jialilian5181/article/details/78445038

版权

python 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

系统环境为ubuntu16.04

编译器为python2.7

模块中的函数需要添入三个变量，分别为待处理图像的文件夹路径，xml文件夹路径，处理后图像的文件夹路径

from __future__ import division
import os
import xml.dom.minidom
import cv2

def read_xml(ImgPath, AnnoPath, Savepath):

imagelist = os.listdir(AnnoPath)
    for image in imagelist:
        image_pre, ext = os.path.splitext(image)
        imgfile = ImgPath + '/' + image_pre + '.jpg'
        xmlfile = AnnoPath + '/'  + image_pre + '.xml'
        im = cv2.imread(imgfile)
        DomTree = xml.dom.minidom.parse(xmlfile)
        annotation = DomTree.documentElement
        filenamelist = annotation.getElementsByTagName('filename')
        filename = filenamelist[0].childNodes[0].data
        objectlist = annotation.getElementsByTagName('object')
        i = 1
        for objects in objectlist:

namelist = objects.getElementsByTagName('name')
            objectname = namelist[0].childNodes[0].data
            bndbox = objects.getElementsByTagName('bndbox')

for box in bndbox:
                try:
                    x1_list = box.getElementsByTagName('xmin')
                    x1 = int(x1_list[0].childNodes[0].data)
                    y1_list = box.getElementsByTagName('ymin')
                    y1 = int(y1_list[0].childNodes[0].data)
                    x2_list = box.getElementsByTagName('xmax')
                    x2 = int(x2_list[0].childNodes[0].data)
                    y2_list = box.getElementsByTagName('ymax')
                    y2 = int(y2_list[0].childNodes[0].data)

minX = x1
                    minY = y1
                    maxX = x2
                    maxY = y2

if(i % 9 == 0):
                        color = (128,0,0)
                    elif (i % 9 == 1):
                        color = (153, 51, 0)
                    elif (i % 9 == 2):
                        color = (255, 204, 0)
                    elif (i % 9 == 3):
                        color = (0, 51, 0)
                    elif (i % 9 == 4):
                        color = (51, 204, 204)
                    elif (i % 9 == 5):
                        color = (128, 0, 128)
                    elif (i % 9 == 6):
                        color = (0, 255, 255)
                    elif (i % 9 == 7):
                        color = (60, 179, 113)
                    elif (i % 9 == 8):
                        color = (255, 127, 80)
                    elif (i % 9 == 9):
                        color = (255, 0, 0)
                    cv2.rectangle(im,(minX,minY),(maxX,maxY),color,2)

#                    path = Savepath + '/' + image_pre + '_New' + '.jpg'
                    path = Savepath + '/' + image_pre + '.jpg'
                    font = cv2.FONT_HERSHEY_SIMPLEX
                    cv2.putText(im, objectname, (minX,minY - 7), font, 0.5, (0, 0, 255), 1)
                    cv2.imwrite(path, im)

i += 1
                except Exception, e:
                    print e