首页|期刊导航|光学精密工程|基于并联双注意力的轻量级小样本矿石粒度检测

基于并联双注意力的轻量级小样本矿石粒度检测OA

A lightweight few-shot ore detector with parallel dual attention

中文摘要英文摘要

针对传统目标检测方法在矿石粒度检测中存在的计算复杂度高、特征鲁棒性差及分类器性能受限等问题,本文提出一种小样本目标检测算法,旨在降低标注与计算成本,并提升模型在数据稀缺场景下的泛化能力.设计以CenterNet2为框架,采用轻量级VoVNet作为主干网络以保证检测速度;核心创新是设计了一个并联双注意力特征融合模块,其中通道交叉注意力模块用于重校准通道维度特征,空间分组注意力模块聚焦目标关键区域,二者协同增强判别性特征融合能力,从而精准指导查询图像检测.在矿石数据集上进行测试,所提模型平均精度(AP)达55.2%,AP50与AP75分别为78.5%和66.9%,推理速度达57 Frame/s(FPS),注意力模块参数量仅为16.1兆字节(MB),体现出优异的精度-效率均衡性.实验表明该方法能有效提升小样本矿石粒度检测的感知性能,且具备极高的边缘端落地潜力,为解决智能矿山中算力受限条件下的实时检测难题提供了可靠的技术方案.

To address the high computational complexity,limited feature robustness,and constrained clas-sifier performance of conventional object detection methods in ore particle size detection,a few-shot object detection approach was proposed to reduce annotation cost and improve generalization under data-scarce conditions.The proposed method was built upon the CenterNet2 framework and employed a lightweight VoVNet as the backbone to ensure detection efficiency.A parallel dual-attention feature fusion module was designed as the core component.Specifically,a channel cross-attention module was introduced to re-calibrate channel-wise feature responses,while a spatial group-attention module emphasized discriminative target regions.The coordinated operation of the two modules enhanced the fusion of task-relevant features and provided effective guidance for query image detection in few-shot scenarios.Experimental results on an ore dataset show that the proposed model achieved an average precision(AP)of 55.2%,with AP50 and AP75 reaching 78.5%and 66.9%,respectively.The inference speed reached 57 frames per second(FPS),while the attention module required only 16.1 M parameters,indicating a favorable trade-off be-tween accuracy and efficiency.Experimental results demonstrate that the proposed method effectively en-hances the perception performance of few-shot ore particle size detection.Moreover,it possesses high po-tential for edge deployment,providing a reliable technical solution for real-time detection challenges in smart mines under computation-constrained conditions.

孙国栋;刘明轩;李仕宬;吴波

湖北工业大学 机械工程学院,湖北 武汉 430068||湖北工业大学 现代制造质量工程湖北省重点实验室,湖北 武汉 430068湖北工业大学 机械工程学院,湖北 武汉 430068湖北工业大学 底特律绿色工业学院,湖北 武汉 430068中国科学院上海高等研究院,上海 201210

信息技术与安全科学

计算机视觉小样本目标检测轻量化矿石图像实时检测

computer visionfew-shot object detectionlightweightore imagesreal-time

《光学精密工程》 2026 (2)

309-321,13

国家自然科学基金(No.51775177)湖北省揭榜制科技项目(No.2024BEB018)

10.37188/OPE.20263402.0309

评论