计算机科学 ›› 2026, Vol. 53 ›› Issue (10): 189-197.doi: 10.11896/jsjkx.250800084
张用明1,2, 贺祥1, 刘维来3, 徐鹤1
ZHANG Yongming1,2, HE Xiang1, LIU Weilai3, XU He1
摘要: 针对矿石图像中存在微裂隙、矿物杂质、分层裂纹及不规则破碎区域等复杂缺陷的局部纹理特征问题,提出一种融合轻量化主干网络特征提取模块CIR、轻量化颈部结构Slim-neck和损失函数WIoU的改进YOLOv11目标检测模型YOLOv11-CSW。首先,在骨干网络引入反向残差移动块IRMB模块,并使用级联分组注意力模块CGA(Cascaded Group Attention)对其进行改进,降低模型参数量与计算量。其次,引入轻量化网络模块(Slim-neck)对颈部进行改造,在不增加参数成本的前提下,增强多尺度特征融合能力。最后,利用WIoU损失函数精准计算损失并加速边界框回归过程。在矿石数据集上进行的实验表明,与原始 YOLOv11算法相比,改进方法参数量减少6.2%,运算量减少8%,mAP50提升2.9%。所提模型在提高检测精确度的同时也减少了模型参数,实现了模型轻量化和性能的平衡,能够应用于矿石识别实时检测系统中。
中图分类号:
| [1] WANG H Y,LIU Z Y,QU F M,et al.Development of online detection technologies for ore grade [J].Strategic Study of CAE,2024,26(3):152-163. [2] International Energy Agency(IEA).Global Critical MineralsOutlook 2024[R].Paris:International Energy Agency,2024. [3] LIU X B,XIAO X,WANG L,et al.Anchor-free object detection methods and their application progress in complex scenes [J].Acta Automatica Sinica,2023,49(7):1369-1392. [4] XIAO J S,ZHAO T,ZHOU J,et al.Small object detection network based on context augmentation and feature refinement [J].Journal of Computer Research and Development,2023,60(2):465-474. [5] BAUERLE T J,SAMMARCO J J,DUGDALE Z J,et al.Thehuman factors of mineworker fatigue:An overview on prevalence,mitigation,and what’s next[J].American Journal of Industrial Medicine,2022,65(11):832-839. [6] MA X Y,DU X T,CAI Q,et al.Survey on testing of deep learning frameworks [J].Journal of Software,2024,35(8):3752-3784. [7] GIRSHICK R,DONAHUE J,DARRELL T,et al.Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Columbus:IEEE,2014:580-587. [8] REN S Q,HE K M,GIRSHICK R,et al.Faster R-CNN:To-wards Real-Time Object Detection with Region Proposal Networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149. [9] HE K,GKIOXARI G.Mask R-CNN[J].IEEE Transactions on Pattern Analysis & Machine Intelligence,2020,42(2):386-397. [10] LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single ShotMultiBox Detector[C]//Proceedings of the 14th European Conference on Computer Vision(ECCV).Amsterdam:Springer,2016:21-37. [11] REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-Time Object Detection[C]//Proceedings of the IEEE Conference on Computer Visionand Pattern Recognition(CVPR).Las Vegas:IEEE,2016:779-788. [12] LIN T Y,GOYAL P,GIRSHICK R,et al.Focal Loss for Dense Object Detection[C]//Proceedings of the IEEE International Conference on Computer Vision(ICCV).Venice:IEEE,2017:2980-2988. [13] HE W X,JING H D,LIU X B,et al.Research on iron ore grade identification technology based on YOLOv4-tiny [J].Metal Mine,2021(10):150-154. [14] QIU Z,HUANG X,XU X.Minima-YOLO:A Lightweight Identification Method for Lithium Mineral Components Under a Microscope Based on YOLOv8[J].Sensors,2025,25(7):2048. [15] TANG Y,LIU Z S,TAO S L.Ore object detection algorithm based on improved YOLOv5 [J].Journal of Sensor Technology and Application,2023,11(6):477-484. [16] TAO H,ZHANG L,SUN Z,et al.Multi-Strategy Improvement of Coal Gangue Recognition Method of YOLOv11[J].Sensors,2025,25(7):1983. [17] GAO L,YU P,DONG H,et al.Multi-Scale Fusion Lightweight Target Detection Method for Coal and Gangue Based on EMBS-YOLOv8s[J].Sensors,2025,25(6):1734. [18] ZONG G,YUE Y,SHAN W.Optimization Study of CoalGangue Detection in Intelligent Coal Selection Systems Based on the Improved Yolov8n Model[J].Electronics,2024,13(21):4155. [19] WU L,HAO H Y,SONG Y.A survey on industrial metal surface defect detection based on computer vision [J].Acta Automatica Sinica,2024,50(7):1261-1283. [20] JIANG Z T,ZHAI F S,QIAN Y,et al.Low-illumination object detection combining feature enhancement and multi-scale receptive field [J].Journal of Computer Research and Development,2023,60(4):903-915. [21] MU Y D,LI W M,FAN Z H,et al.Research on dataflow architecture optimization for YOLO neural networks [J].Chinese Journal of Computers,2025,48(1):82-99. [22] RAHIMA K,HUSSAIN M.Yolov11:An overview of the keyarchitectural enhancements[J].arXiv:2410.17725,2024. [23] ZHOU Y,HAN B,GAO X B,et al.Domain adaptive object detection based on attention mechanism and cycle domain triplet loss [J].Acta Automatica Sinica,2024,50(11):2188-2203. [24] ZHANG J,LI X,LI J,et al.Rethinking Mobile Block for Efficient Attention-based Models[C]//2023 IEEE/CVF International Conference on Computer Vision(ICCV).2023:1389-1400. [25] LIU X,PENG H,ZHENG N,et al.EfficientViT:Memory Efficient Vision Transformer with Cascaded Group Attention[C]//2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).2023:14420-14430. [26] LI H,LI J,WEI H,et al.Slim-neck by GSConv:a lightweight-design for real-time detector architectures[J].Journal of Real-Time Image Processing,2024,21(3):62. [27] HU H J,QIN Y F,XU H,et al.An improved YOLOv8 object detection algorithm for UAV aerial images [J].Computer Science,2025,52(4):202-211. [28] TONG Z,CHEN Y,XU Z,et al.Wise-IoU:bounding box re-gression loss with dynamic focusing mechanism[J].arXiv:2301.10051,2023. |
|
||