在处理faster-rcnn和yolo时笔者使用labelimg标注图片

但是我们只截取了大概800张左右的图,这个量级在训练时肯定是不够的,所以我们需要使用数据增强(无非是旋转加噪调量度)来增加我们的训练样本

这里附上 视频按帧率截取图片的脚本代码  https://www.cnblogs.com/bob-jianfeng/p/11140444.html

网上找到一个非常好的开源代码用来数据增强,这里记录一下,以备以后使用

https://github.com/xinyu-ch/Data-Augment

下载后

只需将里面IMG_dir和XML_dir 修改为我们原始xml路径和存放路径

 IMG_DIR =r"xxxxx"
XML_DIR =r"xxxxx"

默认按照1:20的数量增加

可以修改

 AUGLOOP = 20  # 每张影像增强的数量

修改增强的数量

github不能下的我附上百度云分享

链接:https://pan.baidu.com/s/1ei47dUDuK7zqvRYhK3cwow
提取码:hmui
复制这段内容后打开百度网盘手机App,操作更方便哦

05-18 16:31