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

 

上一篇:JS实现自动轮播图效果(js案例)


下一篇:微信小程序点击图片放大