|
基于融合模型的心脏病图像检测研究
|
Abstract:
针对目前临床心脏病图像检测中存在的准确率低、效果差等问题,提出了一种融合YOLOv5和Attention的检测技术。这种方法旨在充分利用YOLOv5模型在目标检测领域的优势,同时结合Attention注意力机制,以提高对心脏病图像的识别和分类能力。据调查显示,传统的YOLOv5模型在检测心脏病图像时准确率较低且在处理复杂背景和局部特征方面存在一定的局限性,分类准确率不足80%,在实际医疗检测中误差较大。在此基础上,进一步将Attention注意力机制与YOLOv5模型相融合。具体来说,首先通过YOLOv5模型对图像进行特征提取,使用分支网络分别预测特征图中物体的位置和类别,初步得到检测结果并用于下游任务进行分类。然后,Attention机制会根据输入检测结果的特征分布自动计算出每个区域的权重,使模型在检测心脏病图像时能够聚焦于特定的部分,更好地捕捉到关键信息,从而使心脏病图像分类准确性达到98%以上。
A detection technique combining YOLOv5 and Attention is proposed to address the issues of low accuracy and poor effectiveness in current clinical heart disease image detection. This method aims to fully utilize the advantages of the YOLOv5 model in the field of object detection, while combining the Attention mechanism to improve the recognition and classification ability of heart disease images. According to a survey, the traditional YOLOv5 model has low accuracy in detecting heart disease images and has certain limitations in processing complex backgrounds and local features. The classification accuracy is less than 80%, and there is a significant error in actual medical detection. On this basis, the Attention mechanism is further integrated with the YOLOv5 model. Specifically, the YOLOv5 model is first used to extract features from the image, and a branch network is used to predict the position and category of objects in the feature map. The initial detection results are ob-tained and used for downstream task classification. Then, the Attention mechanism will automatically calculate the weight of each region based on the feature distribution of the input detection results, enabling the model to focus on specific parts when detecting heart disease images, better capturing key information, and achieving a classification accuracy of over 98% for heart disease images.
[1] | 朱家群. 支持向量机及其在医学图像可视化中图像分割的应用研究[D]: [硕士学位论文]. 南京: 南京理工大学, 2008. |
[2] | 吴彰良, 孙长库, 刘洁. 基于支持向量机的油封缺陷图像检测方法[J]. 光电工程, 2012, 39(3): 40-45. |
[3] | 张作淳. 随机森林对遥感高分辨率图像沿海地貌检测的应用[J]. 山东工业技术, 2015(12): 278-279.
https://doi.org/10.16640/j.cnki.37-1222/t.2015.12.087 |
[4] | 谢宁. 基于深度学习的生活垃圾图像检测分类方法研究[D]: [硕士学位论文]. 广州: 广州大学, 2023. |
[5] | 赵越, 卫勇, 单慧勇, 等. 基于深度学习的高分辨率麦穗图像检测方法[J]. 中国农业科技导报, 2022, 24(9): 96-105. https://doi.org/10.13304/j.nykjdb.2021.0612 |
[6] | 张桃红, 范素丽, 郭徐徐, 等. 基于数据融合的智能医疗辅助诊断方法[J]. 工程科学学报, 2021, 43(9): 1197-1205.
https://doi.org/10.13374/j.issn2095-9389.2021.01.12.003 |
[7] | 古静, 朱志宇. 改进YOLOv5的红外船舶目标检测算法[J/OL]. 电光与控制, 1-7.
http://kns.cnki.net/kcms/detail/41.1227.TN.20230818.1109.004.html, 2023-08-23. |
[8] | 鲁斌, 刘亚伟, 张宇航, 等. 基于密度感知和自注意力的点云分割算法[J/OL]. 激光与光电子学进展, 1-17.
http://kns.cnki.net/kcms/detail/31.1690.TN.20230821.1429.082.html, 2023-08-23. |
[9] | 何剑. 人耳识别研究综述[J]. 泰州职业技术学院学报, 2009, 9(1): 37-40. |
[10] | 李文博, 赵正旭. 基于YOLOv5的遥感图像小目标检测[J]. 科技创新与应用, 2023, 13(6): 63-67.
https://doi.org/10.19981/j.CN23-1581/G3.2023.06.014 |