【数据集处理】怎样将xml文件信息还原到原始图片中

拿到一批别人标注的图片xml文件想查看别人标注的具体情况将xml里面的坐标名称信息还原到原始图片中。# -*- coding: utf-8 -*-importxml.etree.ElementTreeasETimportos,cv2# 将路径换成自己的即可annota_dirC:/Users/Lenovo/Desktop/VOCdevkit/VOC2007/Annotations#xml文件保存的路径origin_dirC:/Users/Lenovo/Desktop/VOCdevkit/VOC2007/JPEGImages#jpg图片保存的路径target_dir1C:/Users/Lenovo/Desktop/VOCdevkit/VOC2007/remark#生成带有标签信息的文件夹路径即想将转化后的图片保存在哪defdivide_img(oriname):img_fileos.path.join(origin_dir,oriname.jpg)imcv2.imread(img_file)# 读取每个原图像的xml文件xml_fileos.path.join(annota_dir,oriname.xml)treeET.parse(xml_file)roottree.getroot()forobjectinroot.findall(object):object_nameobject.find(name).text Xminint(object.find(bndbox).find(xmin).text)Yminint(object.find(bndbox).find(ymin).text)Xmaxint(object.find(bndbox).find(xmax).text)Ymaxint(object.find(bndbox).find(ymax).text)color(4,250,7)cv2.rectangle(im,(Xmin,Ymin),(Xmax,Ymax),color,2)fontcv2.FONT_HERSHEY_SIMPLEX cv2.putText(im,object_name,(Xmin,Ymin-7),font,0.5,(6,230,230),2)img_nameoriname.jpgto_nameos.path.join(target_dir1,img_name)cv2.imwrite(to_name,im)img_listos.listdir(origin_dir)fornameinimg_list:divide_img(name.rstrip(.jpg))print(转化完成)转化结果如图