欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python将xml文件数据增强(labelimg)

程序员文章站 2022-04-15 15:15:20
在处理faster-rcnn和yolo时笔者使用labelimg标注图片 但是我们只截取了大概800张左右的图,这个量级在训练时肯定是不够的,所以我们需要使用数据增强(无非是旋转加噪调量度)来增加我们的训练样本 这里附上 视频按帧率截取图片的脚本代码 https://www.cnblogs.com/ ......

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

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

这里附上 视频按帧率截取图片的脚本代码  

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

https://github.com/xinyu-ch/data-augment

下载后

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

1 img_dir =r"xxxxx"
2 xml_dir =r"xxxxx"

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

可以修改

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

修改增强的数量

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

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