在处理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,操作更方便哦