计算机科学 ›› 2026, Vol. 53 ›› Issue (10): 189-197.doi: 10.11896/jsjkx.250800084

• 计算机图形学 & 多媒体 • 上一篇    下一篇

YOLOv11-CSW:轻量化多尺度矿石纹理缺陷检测模型

张用明1,2, 贺祥1, 刘维来3, 徐鹤1   

  1. 1 南京邮电大学计算机学院、软件学院、网络空间安全学院 南京 210023
    2 安徽新华学院大数据与人工智能学院 合肥 230088
    3 中国科学技术大学工程科学学院 合肥 230027
  • 收稿日期:2025-08-18 修回日期:2025-10-27 发布日期:2026-10-13
  • 通讯作者: 徐鹤(xuhe@njupt.edu.cn)
  • 作者简介:(zyming@njupt.edu.cn)
  • 基金资助:
    江苏省研究生实践创新计划项目(SJCX24_0315)

YOLOv11-CSW:Lightweight Multi-scale Ore Texture Defect Detection Model

ZHANG Yongming1,2, HE Xiang1, LIU Weilai3, XU He1   

  1. 1 School of Computer Science,Nanjing University of Posts and Telecommunications,Nanjing 210023, China
    2 College of Big Data and Artificial Intelligence,Anhui Xinhua University,Hefei 230088, China
    3 School of Engineering Sciences,University of Science and Technology of China,Hefei 230027, China
  • Received:2025-08-18 Revised:2025-10-27 Online:2026-10-13
  • About author:ZHANG Yongming,born in 1976,Ph.D,senior engineer,is a member of CCF(No.15973G).His main research in-terests include artificial intelligence and information security.
    XU He,born in 1985,Ph.D,professor,master’s supervisor,is a member of CCF(No.19957S).His main research interests include artificial intelligence,big data and Internet of Things techno-logy.
  • Supported by:
    Jiangsu Provincial Graduate Practice and Innovation Program Project(SJCX24_0315).

摘要: 针对矿石图像中存在微裂隙、矿物杂质、分层裂纹及不规则破碎区域等复杂缺陷的局部纹理特征问题,提出一种融合轻量化主干网络特征提取模块CIR、轻量化颈部结构Slim-neck和损失函数WIoU的改进YOLOv11目标检测模型YOLOv11-CSW。首先,在骨干网络引入反向残差移动块IRMB模块,并使用级联分组注意力模块CGA(Cascaded Group Attention)对其进行改进,降低模型参数量与计算量。其次,引入轻量化网络模块(Slim-neck)对颈部进行改造,在不增加参数成本的前提下,增强多尺度特征融合能力。最后,利用WIoU损失函数精准计算损失并加速边界框回归过程。在矿石数据集上进行的实验表明,与原始 YOLOv11算法相比,改进方法参数量减少6.2%,运算量减少8%,mAP50提升2.9%。所提模型在提高检测精确度的同时也减少了模型参数,实现了模型轻量化和性能的平衡,能够应用于矿石识别实时检测系统中。

关键词: 矿石图像, 骨干网络模块, 特征融合网络, 损失函数, 轻量化

Abstract: To address the challenge of local texture features in mineral images containing complex defects such as micro-fractures,mineral impurities,layered cracks,and irregularly fractured regions,this paper proposes an improved YOLOv11 object detection model named YOLOv11-CSW.This model integrates lightweight backbone network feature extraction module CIR,lightweight neck structure Slim-neck,and WIoU(Weighted Intersection over Union ) loss function.Firstly,an IRMB(Inverted Resi-dual Mobile Block) module is introduced into the backbone network and enhanced with a CGA(Cascaded Group Attention) mo-dule to reduce the number of model parameters and computational cost.Secondly,a lightweight Slim-neck module is incorporated to redesign the neck part,improving multi-scale feature fusion capabilities while maintaining parameter efficiency.Finally,the WIoU loss function is employed to accurately compute the loss and accelerate the bounding box regression process.Compared with the original YOLOv11 algorithm,experiments on an ore dataset show that the improved method reduces the number of parameters by 6.2% and computational load by 8%,while increasing mAP50 by 2.9%.The proposed model not only improves detection accuracy but also reduces model parameters,achieving a balance between lightweight design and performance,making it suitable for real-time ore recognition and detection systems.

Key words: Ore image, Backbone network module, Feature fusion network, Loss function, Lightweight

中图分类号: 

  • TP391
[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.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!