0
点赞
收藏
分享

微信扫一扫

YoloV8提升训练速度

 

import imgaug.augmenters as iaa
import cv2
import numpy as np
# 定义数据增强方法
seq = iaa.Sequential([
    # iaa.Flipud(0.5),  # 纵向翻转,概率为50%
    # iaa.Fliplr(0.5),  # 横向翻转,概率为50%
    # iaa.Affine(rotate=(-15, 15)),  # 旋转,范围在-15到15度之间
    iaa.GammaContrast(gamma=(0.8, 1.2)),  # 伽马校正,范围在0.5到1.5之间
    iaa.AdditiveGaussianNoise(scale=0.05 * 255),  # 添加高斯噪声
    iaa.Crop(px=(0, 16)),  # 随机裁剪,每侧裁剪0到16像素
    iaa.GaussianBlur(sigma=(0, 3.0)),  # 高斯模糊,sigma范围在0到3.0之间
    iaa.ElasticTransformation(alpha=(0, 30), sigma=5),  # 弹性扭曲,alpha范围在0到30,sigma为5
    iaa.Affine(translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)})  # 平移,x和y方向上平移-10%到10%
])
# 定义图像特有的增强方法
s
举报

相关推荐

0 条评论