arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.15918eess.IVcs.CV

应用卷积神经网络、ResNet和Grad-CAM的黄斑变性感知与定位

Perception and Localization of Macular Degeneration Applying Convolutional Neural Network, ResNet and Grad-CAM

Rajshahi University of Engineering & Technology(拉杰沙希工程技术大学)

Tahmim Hossain, Sagor Chandro Bakchy

AI总结:

本研究利用CNN及多种ResNet骨干网络对健康与黄斑变性眼底图像进行分类,其中ResNet50骨干模型在90%训练集划分下准确率达98.7%,并通过Grad-CAM实现病变区域定位。

摘要:

黄斑变性是一种众所周知的视网膜疾病,会使患者出现视力模糊。本研究通过定位眼底的受影响区域,对健康眼底和黄斑变性眼底进行分类。使用一种CNN架构以及以CNN结合ResNet架构(ResNet50、ResNet50v2、ResNet101、ResNet101v2、ResNet152、ResNet152v2)作为骨干网络,对两类眼底图像进行分类。数据按三种方式划分:(a)训练集90%,测试集10%;(b)训练集80%,测试集20%;(c)训练集50%,测试集50%。训练完成后,根据评估指标选择最佳模型。在这些模型中,以ResNet50为骨干的CNN表现最佳,在90%训练集和10%测试集的数据划分下,训练准确率达到98.7%。利用该模型,我们进行了Grad-CAM可视化,以获取眼底受影响区域的定位。

英文摘要:

A well-known retinal disease that sends blurry visions to the affected patients is Macular Degeneration. This research is based on classifying the healthy and macular degeneration fundus by localizing the affected region of the fundus. A CNN architecture and CNN with ResNet architecture (ResNet50, ResNet50v2, ResNet101, ResNet101v2, ResNet152, ResNet152v2) as the backbone are used to classify the two types of fundus. The data are split into three categories including (a) Training set is 90% and Testing set is 10% (b) Training set is 80% and Testing set is 20%, (c) Training set is 50% and Testing set is 50%. After the training, the best model has been selected from the evaluation metrics. Among the models, CNN with a backbone of ResNet50 performs best which gives the training accuracy of 98.7% for 90% train and 10% test data split. With this model, we have performed the Grad-CAM visualization to get the region of the affected area of the fundus.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑