Chen Peng, Liu Songyan, Lu Wenbin, Lu Fangpeng, Ding Boyang
Heilongjiang University, Harbin, 150080, China.
Sci Rep. 2024 Nov 4;14(1):26702. doi: 10.1038/s41598-024-77878-6.
The WCAY (weighted channel attention YOLO) model, which is meticulously crafted to identify fracture features across diverse X-ray image sites, is presented herein. This model integrates novel core operators and an innovative attention mechanism to enhance its efficacy. Initially, leveraging the benefits of dynamic snake convolution (DSConv), which is adept at capturing elongated tubular structural features, we introduce the DSC-C2f module to augment the model's fracture detection performance by replacing a portion of C2f. Subsequently, we integrate the newly proposed weighted channel attention (WCA) mechanism into the architecture to bolster feature fusion and improve fracture detection across various sites. Comparative experiments were conducted, to evaluate the performances of several attention mechanisms. These enhancement strategies were validated through experimentation on public X-ray image datasets (FracAtlas and GRAZPEDWRI-DX). Multiple experimental comparisons substantiated the model's efficacy, demonstrating its superior accuracy and real-time detection capabilities. According to the experimental findings, on the FracAtlas dataset, our WCAY model exhibits a notable 8.8% improvement in mean average precision (mAP) over the original model. On the GRAZPEDWRI-DX dataset, the mAP reaches 64.4%, with a detection accuracy of 93.9% for the "fracture" category alone. The proposed model represents a substantial improvement over the original algorithm compared to other state-of-the-art object detection models. The code is publicly available at https://github.com/cccp421/Fracture-Detection-WCAY .
本文提出了WCAY(加权通道注意力YOLO)模型,该模型经过精心设计,用于识别不同X射线图像部位的骨折特征。该模型集成了新颖的核心算子和创新的注意力机制,以提高其效能。首先,利用动态蛇形卷积(DSConv)善于捕捉细长管状结构特征的优势,我们引入DSC-C2f模块,通过替换部分C2f来增强模型的骨折检测性能。随后,我们将新提出的加权通道注意力(WCA)机制集成到架构中,以加强特征融合并改善不同部位的骨折检测。进行了对比实验,以评估几种注意力机制的性能。这些增强策略通过在公共X射线图像数据集(FracAtlas和GRAZPEDWRI-DX)上的实验得到了验证。多次实验比较证实了该模型的效能,展示了其卓越的准确性和实时检测能力。根据实验结果,在FracAtlas数据集上,我们的WCAY模型的平均精度均值(mAP)比原始模型显著提高了8.8%。在GRAZPEDWRI-DX数据集上,mAP达到64.4%,仅“骨折”类别的检测准确率就达到93.9%。与其他先进的目标检测模型相比,所提出的模型相对于原始算法有了实质性的改进。代码可在https://github.com/cccp421/Fracture-Detection-WCAY上公开获取。