2.显示xml标注框

最新推荐文章于 2024-01-05 08:41:02 发布

陈路飞

最新推荐文章于 2024-01-05 08:41:02 发布

阅读量398

点赞数

CC 4.0 BY-SA版权

分类专栏：脚本

本文链接：https://blog.csdn.net/chenlufei_i/article/details/114879945

脚本专栏收录该内容

3 篇文章

订阅专栏

import os
import xml.dom.minidom
import cv2
a=[]
ImgPath = 'C:\\Users\chenlufei\Desktop\data\stair\stair\\'
AnnoPath = 'C:\\Users\chenlufei\Desktop\data\stair\stair_xml\outputs\\'

#ImgPath = 'C:\\Users\chenlufei\Desktop\data\stair\stair\\'
#AnnoPath = 'C:\\Users\chenlufei\Desktop\data\stair\stair_xml\outputs\\'
imagelist = os.listdir(ImgPath)

for image in imagelist:
image_pre, ext = os.path.splitext(image) # 分离文件名与扩展名
imgfile = ImgPath + image
print(imgfile)
xmlfile = AnnoPath + image_pre + '.xml'

# 打开xml文档
DOMTree = xml.dom.minidom.parse(xmlfile)
# 得到文档元素对象
collection = DOMTree.documentElement

# 读取图片
img = cv2.imread(imgfile)

filenamelist = collection.getElementsByTagName("name")

try:
filename = filenamelist[0].childNodes[0].data
except:
a.append(imgfile)
continue

cv2.putText(img,filename,(10,20), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
img_size_list=collection.getElementsByTagName('size')
for img_size in img_size_list:
img_width_list = img_size.getElementsByTagName('width')
img_height_list = img_size.getElementsByTagName('height')
img_width = img_width_list[0].childNodes[0].data
img_height = img_height_list[0].childNodes[0].data
cv2.putText(img, img_width, (10, 40), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
cv2.putText(img, img_height, (10, 60), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)

# 得到标签名为object的信息
objectlist = collection.getElementsByTagName("object")
for objects in objectlist:
# 每个object中得到子标签名为name的信息
namelist = objects.getElementsByTagName('name')
objectname = namelist[0].childNodes[0].data
bndbox = objects.getElementsByTagName('bndbox')
for box in bndbox:
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)
cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), thickness=2)
cv2.putText(img, objectname, (x1, y1), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
print(a)
cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
\