'''根据xml文件找到对应的图片文件
代码说明:
这段小程序的作用就是根据labelimage标注软件生成的xml文件来找到对应的图片。
例如:我们的图片数据有100张,其中有90张图片里有我们想标注的目标,这样的话我们就会生成90个xml文件,
其中这90个xml文件的名字和那90张有目标的图片的名字一样,我们需要根据这90个xml文件的名字从那100张图片里找到对应的90张图片。
---------------------
作者:生命的呼喊
来源:CSDN
原文:https://blog.csdn.net/ITpfzl/article/details/90082355
版权声明:本文为博主原创文章,转载请附上博文链接!'''
import os
import cv2
input_pic_path = "D:/PointerImg/datasets/"
input_xml_path = "D:/PointerImg/annotations/"
out_pic = "D:/PointerImg/datasets1/"
list_pic = os.listdir(input_pic_path)
for xml_file_name in os.listdir(input_xml_path):
pic_file_name = os.path.splitext(xml_file_name)[0] + ".jpg"
pic_file_path = os.path.join(input_pic_path,pic_file_name)
image = cv2.imread(pic_file_path)
cv2.imwrite(out_pic + pic_file_name, image)
根据xml文件找到对应的图片文件
最新推荐文章于 2023-04-06 17:41:00 发布