Evaluation of multi-classification method of color fundus photograph quality based on ResNet50-OC
10.3760/cma.j.cn115989-20200107-00011
- VernacularTitle:基于ResNet50-OC模型的彩色眼底照片质量多分类方法效果评估
- Author:
Cheng WAN
1
;
Xueting ZHOU
;
Qijing YOU
;
Jianxin SHEN
;
Qiuli YU
Author Information
1. 南京航空航天大学电子信息工程学院 211106
- Keywords:
Artificial intelligence;
Image quality classification;
Convolutional neural network;
One-Cycle learning;
Color fundus photograph
- From:
Chinese Journal of Experimental Ophthalmology
2021;39(9):785-790
- CountryChina
- Language:Chinese
-
Abstract:
Objective:To evaluate the efficiency of ResNet50-OC model based on deep learning for multiple classification of color fundus photographs.Methods:The proprietary dataset (PD) collected in July 2018 in BenQ Hospital of Nanjing Medical University and EyePACS dataset were included.The included images were classified into five types of high quality, underexposure, overexposure, blurred edges and lens flare according to clinical ophthalmologists.There were 1 000 images (800 from EyePACS and 200 from PD) for each type in the training dataset and 500 images (400 from EyePACS and 100 from PD) for each type in the testing dataset.There were 5 000 images in the training dataset and 2 500 images in the testing dataset.All images were normalized and augmented.The transfer learning method was used to initialize the parameters of the network model, on the basis of which the current mainstream deep learning classification networks (VGG, Inception-resnet-v2, ResNet, DenseNet) were compared.The optimal network ResNet50 with best accuracy and Micro F1 value was selected as the main network of the classification model in this study.In the training process, the One-Cycle strategy was introduced to accelerate the model convergence speed to obtain the optimal model ResNet50-OC.ResNet50-OC was applied to multi-class classification of fundus image quality.The accuracy and Micro F1 value of multi-classification of color fundus photographs by ResNet50 and ResNet50-OC were evaluated.Results:The multi-classification accuracy and Micro F1 values of color fundus photographs of ResNet50 were significantly higher than those of VGG, Inception-resnet-v2, ResNet34 and DenseNet.The accuracy of multi-classification of fundus photographs in the ResNet50-OC model was 98.77% after 15 rounds of training, which was higher than 98.76% of the ResNet50 model after 50 rounds of training.The Micro F1 value of multi-classification of retinal images in ResNet50-OC model was 98.78% after 15 rounds of training, which was the same as that of ResNet50 model after 50 rounds of training.Conclusions:The proposed ResNet50-OC model can be accurate and effective in the multi-classification of color fundus photograph quality.One-Cycle strategy can reduce the frequency of training and improve the classification efficiency.
