多标签json的线下增强使用imgaug

1.导入所需要的包

import imgaug.augmenters as iaa
from imgaug.augmentables.kps import KeypointsOnImage, Keypoint
import json
import numpy as np
import cv2
import os
import random

 2.定义自己的文件路径

input_folder = r''  # 替换为输入文件夹路径
output_folder = r''  # 替换为保存增强结果的输出文件夹路径

3.解决cv2不能读取中文路径 

def imread(image_path, flags=cv2.IMREAD_COLOR):
    # 使用cv2.imdecode读取图像，支持中文路径
    return cv2.imdecode(np.fromfile(image_path, dtype=np.uint8), flags)

4.定义数据增强操作（可以按照自己的需求制定增强规则）

# 定义数据增强操作
augmenters = iaa.OneOf([
    iaa.Fliplr(0.5),
    iaa.Flipud(0.5)
    # 左右翻转 100% 概率
    # 其他增强操作可以在此处添加
])

#https://imgaug.readthedocs.io/en/latest/index.html官方文档

5.读取图片和json和相应关键点

# 获取所有