目标检测数据增强:根据寻找相同名称不同后缀的照片

本示例的作用为找到.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)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值