基于改进YOLOv12算法的MRI图像脑肿瘤检测研究OA
Research on MRI brain tumor detection based on improved YOLOv12 algorithm
目的:为了提升原始YOLOv12算法在MRI图像脑肿瘤检测中的性能,提出基于C3k2小波变换卷积模块(C3k2 wavelet transform convolution,C3k2_WTConv)和空间-深度卷积模块(space-to-depth convolution,SPDConv)的改进YOLOv12算法.方法:以原始YOLOv12算法为基础进行改进,将原始YOLOv12算法的主干网络(Backbone)及颈部模块(Neck)中的C3k2模块替换成C3k2_WTConv模块,并将Backbone及Neck中的部分标准Conv模块替换成SPDConv模块,构建YOLOv12-C3k2_WTConv-SPDConv算法.使用Kaggle平台公开发布的MRI脑肿瘤数据集对改进后的算法开展训练与性能验证,并与原始YOLOv12算法以及单次多框检测器(single shot multibox detector,SSD)、实时检测变换器大模型(real-time detection transformer-large variant,RT-DETR-L)算法的各项性能进行对比分析.结果:在MRI图像脑肿瘤检测中,YOLOv12-C3k2_WTConv-SPDConv算法在交并比阈值为0.5时的平均精度均值mAP50与交并比阈值为0.5~0.95时的平均精度均值mAP50-95分别为0.947和0.748,参数量和推理速度分别为2.08M、90帧/s,均优于原始YOLOv12算法;与SSD、RT-DETR-L算法相比,YOLOv12-C3k2_WTConv-SPDConv算法综合性能较优.结论:YOLOv12-C3k2_WTConv-SPDConv算法在MRI图像脑肿瘤检测中综合性能表现较优,能够提升原始YOLOv12算法在MRI图像脑肿瘤检测中的性能,对临床脑肿瘤的早期筛查与诊断具有重要作用.
Objective To propose an improved YOLOv12 algorithm based on the C3k2 wavelet transform convolution module(C3k2_WTConv)and the space-to-depth convolution module(SPDConv),so as to enhance the performance of the original YOLOv12 algorithm for MRI image-based brain tumor detection.Methods An YOLOv12-C3k2_WTConv-SPDConv algorithm was built based on the original YOLOv12 algorithm with the C3k2 module from the Backbone and Neck replaced by the C3K2_WTConv module and some standard Conv modules substituted by the SPDConv module.The proposed algorithm was trained and evaluated for performance using the publicly available MRI brain tumor dataset on the Kaggle platform,which was compared with the original YOLOv12 algorithm,the single shot multibox detector(SSD)algorithm and the real-time detection transformer-large variant(RT-DETR-L)algorithm in terms of various performance metrics.Results When used for MRI image-based brain tumor detection,the YOLOv12-C3k2_WTConv-SPDConv algorithm gained advantages over the original YOLOv12 algorithm,which had the mean average precision at intersection over union threshold 0.5(mAP50)being 0.947,the mean average precision at intersection over union threshold 0.5-0.95(mAP50-95)being 0.748,the Params being 2.08M and the inference speed being 90 frames per second;the YOLOv12-C3k2_WTConv-SPDConv algorithm behaved better than the SSD and RT-DETR-L algorithms.Conclusion The YOLOv12-C3k2_WTConv-SPDConv algorithm functions well in MRI image-based brain tumor detection,enhances the original YOLOv12 algorithm in MRI image-based brain tumor detection and contributes to early screening and diagnosis of brain tumors.[Chinese Medical Equipment Journal,2026,47(3):9-17]
洪成坤;付丽媛
福建中医药大学福总教学医院(第九○○医院)放射诊断科,福州 350025福建中医药大学福总教学医院(第九○○医院)放射诊断科,福州 350025
医药卫生
YOLOv12C3k2_WTConvSPDConvMRI图像脑肿瘤
YOLOv12C3k2_WTConvSPDConvMRI imagebrain tumor
《医疗卫生装备》 2026 (3)
9-17,9
福建省科技计划项目(2021I0037)
评论