对原始图像，xml，label进行水平翻转，进行数据增强

番茄就要炒鸡蛋

于 2021-06-02 21:24:08 发布

阅读量629

点赞数

CC 4.0 BY-SA版权

分类专栏： opencv 文章标签：视觉检测 python

本文链接：https://blog.youkuaiyun.com/hejunran/article/details/117483650

该博客主要围绕Python实现视觉检测展开，虽未给出具体内容，但可知重点在于利用Python达成视觉检测效果，属于信息技术中视觉检测与Python编程结合的范畴。

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

import numpy as np
import cv2
import xml.etree.ElementTree as ET
from PIL import Image

name_classes = ['hqc']  # 类别名，可以更改为对应的voc类别名称即可

def flip_horizontal(jpg_file,new_jpg,png_file,new_png):
    img = Image.open(jpg_file)
    png =Image.open(png_file)
    # out = im.transpose(Image.ROTATE_180)
    out_img = img.transpose(Image.FLIP_LEFT_RIGHT)
    out_png = png.transpose(Image.FLIP_LEFT_RIGHT)

    out_img.save(new_jpg)
    out_png.save(new_png)


def flip_xml(xml_file,new_xml):
    tree = ET.parse(xml_file)
    objs = tree.findall('object')
    for ix, obj in enumerate(objs):
        name = obj.find('name').text
        if name == 'hqc':           # 因为本数据集只针对某一个更改的，你要改的有很多，此处要改为 if name in name_classes :
            print(xml_file)
            obj_new = obj.find('bndbox')
            xmin = str( 256 - int(obj_new.find('xmax').tex