医疗卫生装备2025，Vol.46Issue(5)：1-8,8.DOI:10.19745/j.1003-8868.2025079
基于改进YOLOv5算法的脑出血CT图像识别与分割研究
Improved YOLOv5 algorithm-based research on CT image recognition and segmentation for cerebral hemorrhage
摘要
Abstract
Objective To modify the YOLOv5 algorithm with similarity attention mechanism(SimAM)to enhance the recognition and segmentation accuracy of CT images for cerebral hemorrhage.Methods A basic framework was established with a YOLOv5 algorithm consisting of a backbone network(Backbone),a neck module(Neck)and a head module(Head),and then SimAM was introduced at the end of Backbone to form a YOLOv5-Sim-B algorithm and at the end of Neck to construct a YOLOv5-Sim-N algorithm.The YOLOv5-Sim-B and YOLOv5-Sim-N algorithms were trained and validated using the CT image dataset for cerebral hemorrhage publicly available on the Kaggle competition platform,and compared with the traditional YOLOv5 algorithm for recognizing and segmenting cerebral hemorrhagic lesions in CT images.Results In case the value of IoU-T was 0.6,the mean average precision(mAP)was 0.967 for YOLOv5-Sim-B algorithm,0.960 for the YOLOv5-Sim-N algorithm and 0.964 for the traditional YOLOv5 algorithm during the recognition and segmentation of cerebral hemorrhagic lesions in CT images.Conclusion The proposed algorithm gains advantages in detection accuracy and robustness,and can efficiently identify and segment cerebral hemorrhage foci in CT images.[Chinese Medical Equipment Journal,2025,46(5):1-8]关键词
YOLOv5算法/相似性注意力机制/脑出血/CT图像识别/CT图像分割Key words
YOLOv5 algorithm/similarity attention mechanism/cerebral haemorrhage/CT image recognition/CT image segmentation分类
基础医学引用本文复制引用
洪成坤,杨涛,付丽媛..基于改进YOLOv5算法的脑出血CT图像识别与分割研究[J].医疗卫生装备,2025,46(5):1-8,8.基金项目
福建省科技计划项目(2021I0037) （2021I0037）
