针对小目标的目标增强算法

针对小目标的目标增强算法

自己写的数据增强算法,请随意取用,包括椭圆旋转,随机复制。
找了好多并没有找到椭圆旋转和随机复制的代码,于是就自己写了一个,程序中自动生成annotation文件,不需要额外标记。
链接: [link]https://github.com/delixing/Data-Augmentation.

椭圆旋转

椭圆旋转是根据Towards Rotation Invariance in Object Detection这篇论文写的。
https://openaccess.thecvf.com/content/ICCV2021/papers/Kalra_Towards_Rotation_Invariance_in_Object_Detection_ICCV_2021_paper.pdf 通过将旋转的bbox从矩形转换为椭圆形,使得旋转后的bbox更好的拟合目标,从而提高mAP。
在这里插入图片描述
在这里插入图片描述

随机复制

随机复制是根据Augmentation for small object detection这篇论文写的 https://arxiv.org/pdf/1902.07296.pdf 因为在论文中oversample rate最好的情况是3倍,所以程序中用下面的策略进行:
1.首先随机挑选一张图片作为原图。
2.然后从数据集再随机挑选三张图片。
3.将三张图片中的目标拷贝到原图上去。
4.修改annotation文件。
本方法主要针对小目标的数据增强,因为考虑到覆盖的原因,所以请务必使用目标较小的,密度较低的数据集,不然会导致操作时间很长。
在这里插入图片描述

版权声明:本文为CSDN博主「我爱吃桃子」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_38875668/article/details/121240559

我爱吃桃子

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

Detectron2 updating...

介绍 之前一直看相关的api,包括数据集的注册,detectron这种标准数据集的custom注册等源代码分析,对json文件也有了很清晰的认识,COCO的注册更简单一些,直接调用reg

YOLOX训练自己的数据集,txt形式

YOLOX官方支持训练VOC和COCO数据集,但习惯了Yolov3~v5的txt加载数据集,尤其是训练自己的数据集时,标签写入txt文本更方便些,但是YOLOX官方要你自己写&#xff0c

目标检测-分水岭分割与阈值分割对比

1.1 对阴影的处理 基于阈值的分割,是在以某个阈值T,对图像进行分割。分割时由于拍摄时图像阴影处导致分割不理想,结果虽然达到理想预期。但是二值化后的结果以阴影部分外围与图像融合,导致再