'''根据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)