paddledetection中支持不少的数据增强方法,比如GridMask、Cutmix和MixUp等具体可以参考paddle 37 paddledetection中的数据增强方法。但是,缺失裁剪目标然后粘贴的其他数据上的增强方法,还好在https://aistudio.baidu/aistudio/projectdetail/4596890?channelType=0&channel=0 中看到了CopyPaste方法的实现,为此博主对CopyPaste进行扩展实现RotateScaleCopyPaste,支持对裁剪下来的目标进行旋转和缩放,以实现更加复杂的数据扩充方法,核心代码每一行都有注释。

1、基本函数实现

在PaddleDetection/ppdet/data/transform/op_helper.py中添加以下函数

# 旋转angle角度,缺失背景黑色(0, 0, 0)填充
def rotate_scale(image, angle,scale,fill_value=

更多推荐

paddle 41 在paddledetection添加RotateScaleCopyPaste数据增强方法