weixin_42370357的博客

原创 使用imgaug库实现随机方式的图片增强（同时生成新的标签文件【精灵标注助手】）

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