根据XML文件在原始图片上画矩形框

最新推荐文章于 2024-07-20 18:35:22 发布

瀛台夜雪

最新推荐文章于 2024-07-20 18:35:22 发布

阅读量365

点赞数

文章标签： xml python opencv

本文链接：https://blog.csdn.net/sxycylq/article/details/131261053

版权

根据XML文件在原始图片上画矩形框

输入参数：

xml_src：XML文件路径

img_src: 原始图片路径

代码：

def markImage(xml_src,img_src):
    root = ET.parse(xml_src).getroot()
    count = 0

    img = cv2.imread(img_src)
    # 遍历xml文件中object属性
    for obj in root.iter('object'):
        name = obj.find('name').text
        x0 = int(obj.find('bndbox/xmin').text)
        y0 = int(obj.find('bndbox/ymin').text)
        x1 = int(obj.find('bndbox/xmax').text)
        y1 = int(obj.find('bndbox/ymax').text)

        #在图像上绘制矩形框
        cv2.rectangle(img,(x0,y0),(x1,y1),(0,0,255),1)

    cv2.imwrite('res.jpg', img)
    cv2.imshow('image',img)
    cv2.waitKey(0)