机器视觉-YoloV8小目标检测参考文章

本文内容基本摘抄自公众号文章: 小目标检测问题分析和优化思路

小目标检测面临的挑战:

  • 可用特征少
  • 定位精度要求高
  • 数据集中的小目标数量占比少
  • 小目标标注面积占比小
  • 样本不均匀问题
  • 小目标聚集问题
  • 网络结构问题

优化思路1: 数据增强

数据增强是一个提升检测性能简单有效方法, 我们可以使用数据增强方法扩充数据集规模、丰富数据集的多样性,人为提升小目标在整个数据集中的占比,使得网络能充分学习到小目标的特征。
数据增强的方式:

  • 复制粘贴增强(copy paste), 随机将小目标在图像上多复制几个(纯抠图), 或者将复制一块包含多个小目标的区域(带有背景context的抠图),粘贴时可以随机做一些变换(缩放、翻折、旋转等)
  • 过采样增强(Over sampling),同一个图像文件多复制几份。
  • 缩放和拼接(stitching), 将多个图像文件拼接成一个图像文件, yolov8 的 Mosaic 技术就是将4张图像进行翻转\缩放拼成1张图像送到网络中.

增加小目标检测层

使用SAHI技术

https://learnopencv.com/slicing-aided-hyper-inference/

引入注意力机制

解决问题: 目标和背景相似, 目标难以识别.
万字长文解析cv中的注意力机制 https://mp.weixin.qq.com/s/kt3iIuOD3lsZBTIbOSGN0g
https://blog.csdn.net/m0_70388905/article/details/128103103
https://zhuanlan.zhihu.com/p/650309601
https://blog.csdn.net/qq_43471945/category_12467149.html
https://blog.csdn.net/qq_43471945/article/details/132685642

改进特征融合方式

解决问题: 目标尺寸变化多样问题.
https://blog.csdn.net/m0_70388905/article/details/128103103
YOLOv8改进-bifpn https://blog.csdn.net/weixin_45303602/article/details/132135514
YOLOv8主干网络改进 添加加权双向金字塔结构的特征加强模块BiFPN https://zhuanlan.zhihu.com/p/674887344

购买高手的咨询

AI小怪兽 https://zhuanlan.zhihu.com/p/650309601
魔鬼面具 https://blog.csdn.net/qq_37706472
imagegeek https://blog.csdn.net/xss20072754?type=blog

posted @ 2024-02-14 18:48  harrychinese  阅读(722)  评论(0编辑  收藏  举报