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

Classification of Diabetic Retinopathy Severity in Fundus Images with DenseNet121 and ResNet50

Commack High School(康马克高中);Camp Illumina(Illumina 训练营);Glenda Dawson High School(格伦达·道森高中);Mira Loma High School(米拉洛马高中);Ward Melville High School(沃德·梅尔维尔高中)

Jonathan Zhang, Bowen Xie, Xin Wu, Rahul Ram, David Liang

英文摘要:

In this work, deep learning algorithms are used to classify fundus images in terms of diabetic retinopathy severity. Six different combinations of two model architectures, the Dense Convolutional Network-121 and the Residual Neural Network-50 and three image types, RGB, Green, and High Contrast, were tested to find the highest performing combination. We achieved an average validation loss of 0.17 and a max validation accuracy of 85 percent. By testing out multiple combinations, certain combinations of parameters performed better than others, though minimal variance was found overall. Green filtration was shown to perform the poorest, while amplified contrast appeared to have a negligible effect in comparison to RGB analysis. ResNet50 proved to be less of a robust model as opposed to DenseNet121.

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

↑