使用imgaug库实现随机方式的图片增强(同时生成新的标签文件【精灵标注助手】)
前言这段时间在做一个行为识别的项目,数据集是自己拍的视频转换成的RGB图像,标签文件是用精灵标注助手做的标签。由于数据量不够,且为了增加模型训练后的泛化能力,打算做一下数据增强,但是自己精力有限又比较懒,就想着做数据增强的同时也把标签文件自动转换了(也就是说把原图的标签文件中坐标点映射到增强后的图片上,把新的标签点以同样格式保存在xml文件中,按照yolov5的要求,图片名和xml文件名要保持一致),在网上查了很多帖子,没找到合适的方法,只好自己动手写了一个脚本。直接上代码data_aug.py:主
原创
2020-07-29 23:32:32 ·
2037 阅读 ·
1 评论