精细YOLO:基于特征聚合和规范化瓦瑟斯坦距离的简化X射线禁止物体检测网络
Yu-Tong Zhou1, Kai-Yang Cao1, De Li1
1Department of Computer Science and Technology, Yanbian University, Yanji 133002, China.
Sensors (Basel, Switzerland)
|June 19, 2024
概括
这项研究介绍了Fine-YOLO,一种用于X射线安全成像的轻量级物体检测模型. 它以更少的参数实现了高精度,改善了复杂背景中小物体的检测.
科学领域:
- 计算机视觉 计算机视觉
- 人工智能的人工智能
- 图像处理 图像处理
背景情况:
- 由于复杂的背景和小物体,X射线图像对物体检测具有挑战.
- 现有的方法通常需要高的计算成本,阻碍了安全应用的轻量级模型的开发.
研究的目的:
- 提出Fine-YOLO,一种新的轻量级物体检测模型,用于安全任务中的快速和准确的检测.
- 增强特征学习能力,解决现有模型中的信息丢失问题.
主要方法:
- 为YOLOv7骨干设计了一个低参数特征聚合 (LPFA) 结构.
- 引入了高密度特征聚合 (HDFA) 结构,以保存本地细节和深度位置信息.
- 使用规范化瓦瑟斯坦距离 (NWD) 来改善小物体的界限框回归.
主要成果:
- 在EDS数据集中,Fine-YOLO只在1610万个参数中实现了58.3%的检测准确度.
- 在NEU-DET数据集的辅助验证中,检测准确率为73.1%.
结论:
- Fine-YOLO为安全领域的对象检测提供了一个计算效率高,准确的解决方案.
- 该模型的有效性表明,在安全之外的更广泛的检查领域有潜在的应用.


