根据xml文件找到对应的图片文件

find_pic.py

 代码说明：

               这段小程序的作用就是根据labelimage标注软件生成的xml文件来找到对应的图片。

例如：我们的图片数据有100张，其中有90张图片里有我们想标注的目标，这样的话我们就会生成90个xml文件，其中这90个xml文件的名字和那90张有目标的图片的名字一样，我们需要根据这90个xml文件的名字从那100张图片里找到对应的90张图片。

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os
import cv2
input_pic_path = "./Drone_make_picture"
input_xml_path = "./Drone_transform_xml"
out_pic = "./pic_out/"
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)