Yuan Yuan, Minghui Chen, Shuting Ke, Teng Wang, Longxi He, Linjie Lü, Hao Sun, Jiannan Liu. Fundus Image Classification Research Based on Ensemble Convolutional Neural Network and Vision Transformer[J]. Chinese Journal of Lasers, 2022, 49(20): 2007205
- Chinese Journal of Lasers
- Vol. 49, Issue 20, 2007205 (2022)
Fig. 1. Overview of Vit model
Fig. 2. Structure of MBConv and Fused-MBConv. (a) MBConv; (b) Fused-MBConv
Fig. 3. SimAM 3D attention weights
Fig. 4. Confusion matrix of Vit and EfficientNetV2-S models. (a) Vit; (b) EfficientNetV2-S
Fig. 5. Heatmap of abnormal fundus images in dataset. (a) Abnormalities in optic disc; (b) abnormalities in macular
|
Table 1. EfficientNetV2-S architecture
|
Table 2. Fundus dataset
|
Table 3. Accuracy, precision, and specificity of Vit, EfficientNetV2-S, and EfficientNet-Vit models
|
Table 4. Comparison of accuracy indexes of different models
|
Table 5. Comparison of accuracy indexes of models with different weighted factors
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20