目标检测训练单通道图片，批量变三通道，批量修改xml中的depth

最新推荐文章于 2023-05-17 00:15:00 发布

天下第一猛男

最新推荐文章于 2023-05-17 00:15:00 发布

阅读量396

点赞数

文章标签：深度学习 python

本文链接：https://blog.csdn.net/m0_56551038/article/details/117515345

版权

该博客主要介绍了如何在目标检测训练中将单通道图片转换为三通道，并详细说明了批量修改XML文件中depth属性的过程，同时解决了图片保存路径的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
import cv2 as cv

def change_channel(path):

    for file in os.listdir(path):
        pic_file = path + file

        src = cv.imread(pic_file)
        print(src)
        gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY)
        print(gray)

        image = cv.merge([gray, gray, gray])
        print(image)

        cv.imwrite(file,image)

        # break #循环一次做实验，调试代码

if __name__ == '__main__':

    change_channel('F:\BS_IMAGE\demo/')#图片文件夹路径

改变通道数：实际是灰度图复制了三遍，一些算法要求输入三通道。遇到问题：不能保存图片至指定文件夹，只能保存当前目录再剪切

from xml.etree import ElementTree as ET
import os.path


def chang_depth(xml_path):

    for filename in os.listdir(xml_path):
        et = ET.parse(xml_path + filename)  # 解析文件
        root = et.getroot()

        # 一层一层获取
        size = root.find('size')
        depth = size.find