本示例的作用为找到.xml文件对应的.jpg文件
import os
import cv2
input_pic_path = r"D:\Catalogue\master\generate_data\mitios\VOCdevkit2007\VOC2007\JPEGImages"
input_xml_path = r"D:\Catalogue\master\generate_data\mitios\VOCdevkit2007\VOC2007\Annotations"
def mkdir(path):
# 去除首位空格
path = path.strip()
# 去除尾部 \ 符号
path = path.rstrip("\\")
# 判断路径是否存在
# 存在 True
# 不存在 False
isExists = os.path.exists(path)
# 判断结果
if not isExists:
# 如果不存在则创建目录
# 创建目录操作函数
os.makedirs(path)
print(path + ' 创建成功')
return True
else:
# 如果目录存在则不创建,并提示目录已存在
print(path + ' 目录已存在')
return False
out_pic = r"D:\Catalogue\master\generate_data\mitios_withxml\VOCdevkit2007\VOC2007\JPEGImages"
mkdir(out_pic)
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(os.path.join(out_pic, pic_file_name), image)