光学精密工程2026,Vol.34Issue(2):309-321,13.DOI:10.37188/OPE.20263402.0309
基于并联双注意力的轻量级小样本矿石粒度检测
A lightweight few-shot ore detector with parallel dual attention
摘要
Abstract
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.关键词
计算机视觉/小样本目标检测/轻量化/矿石图像/实时检测Key words
computer vision/few-shot object detection/lightweight/ore images/real-time分类
信息技术与安全科学引用本文复制引用
孙国栋,刘明轩,李仕宬,吴波..基于并联双注意力的轻量级小样本矿石粒度检测[J].光学精密工程,2026,34(2):309-321,13.基金项目
国家自然科学基金(No.51775177) (No.51775177)
湖北省揭榜制科技项目(No.2024BEB018) (No.2024BEB018)