Lian Chaoming, Zhong Shuncong, Zhang Tianfu, Zhou Ning, Xie Maosong. Transfer Learning-Based Classification of Optical Coherence Tomography Retinal Images[J]. Laser & Optoelectronics Progress, 2021, 58(1): 117002
- Laser & Optoelectronics Progress
- Vol. 58, Issue 1, 117002 (2021)
Fig. 1. Global average pooling function for image feature extraction
Fig. 2. OCT images of diabetic retinopathy. (a) Choroidal neovascularization; (b) diabetic macular edema; (c) drusen; (d) normal retina
Fig. 3. Images before and after denoising
Fig. 4. GAP transfer learning network model structure
Fig. 5. Training set curves of control experimental networks. (a)(d) Training set curves of direct transfer learning group; (b)(e) training set curves of fine-tuning transfer learning group; (c)(f) training set curves of GAP transfer learning group
Fig. 6. ROC curve of GAP transfer learning group network classification result. (a) InceptionV3 network; (b) VGG19 network; (c) ResNet50 network
| ||||||||||||||||||||||||||||||||||
Table 1. Retina data set partition
| |||||||||||||||
Table 2. Test accuracy of different pre-training models in control experiment
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20