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

系统环境为ubuntu16.04

编译器为python2.7

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

from __future__ import divisionimport osimport xml.dom.minidomimport cv2def 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