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)