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

最新推荐文章于 2024-12-01 07:30:00 发布

原创

最新推荐文章于 2024-12-01 07:30:00 发布 · 657 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#视觉检测 #python

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

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').text))     # 因为进行水平翻转，这只改变x坐标值，y坐标值不变
            xmax = str( 256 - int(obj_new.find('xmin').text))     # 此处256是图片的宽和高，水平翻转，