Acta Optica Sinica, Volume. 41, Issue 22, 2210002(2021)

Recognition and Classification of Diabetic Retinopathy Based on Improved DR-Net Algorithm

Wen Zheng, Qihao Shen, and Jia Ren*
Author Affiliations
  • School of Mechanical Engineering and Automation, Zhejiang Sci-Tech University, Hangzhou, Zhejiang 310018, China
  • show less
    Figures & Tables(15)
    Example of 2D color fundus
    Examples of DR grading sample
    Original image and RGB channel output image
    Image preprocessing process and means. (a) Original image; (b) filter denoising; (c) contrast enhancement;(d) light equalization
    Architecture of improved DR-Net network model
    Dilated convolution under different expansion rates. (a) d=1; (b) d=2; (c) d=3
    Architecture of aggregated residual
    Structure diagram of attention mechanism
    Examples of improved DR-Net test results
    Accuracy and error curve in training process of improved DR-Net. (a) Accuracy; (b) train loss
    Confusion matrix test results of improved DR-Net model
    Comparison of confusion matrix among mainstream models on DR classification. (a) Densenet121;(b) Mobilenetv2; (c) Efficient-net b5; (d) ResNext50
    Comparison of performance on DR-Net and improved DR-Net models
    • Table 1. Dataset correction and distribution

      View table

      Table 1. Dataset correction and distribution

      DatasetNo_DRMild_DRModerate_DRSevere_DRProliferate_DR
      APTOS 2019413962624976
      Data cleaning363901924172
      Enhanced & extraction1089720768672864
      Eye-PACS2581024435292873708
      Data cleaning2558423175040554686
      Enhanced & extraction23252317252022192058
      Proposed dataset34143037328828912922
    • Table 2. Comparison results of existing DR classification algorithms

      View table

      Table 2. Comparison results of existing DR classification algorithms

      Related researchSensitivitySpecificityPPVNPVAccuracy
      Densenet12182.2583.2883.6084.5983.36
      Mobilenetv287.2287.9190.4687.9888.06
      Ref. [26]97.1186.0393.4196.6392.01
      Ref. [27]97.7287.1189.1797.4592.41
      Efficient-net b592.3992.9192.5292.6292.55
      Ref. [28]84.8090.4092.7292.9292.95
      Ref. [29]90.5095.6094.1790.9696.90
      Ref. [30]95.6098.7098.6695.7397.20
      Improved DR-Net97.9198.0397.9098.0398.04
    Tools

    Get Citation

    Copy Citation Text

    Wen Zheng, Qihao Shen, Jia Ren. Recognition and Classification of Diabetic Retinopathy Based on Improved DR-Net Algorithm[J]. Acta Optica Sinica, 2021, 41(22): 2210002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 25, 2021

    Accepted: Jun. 3, 2021

    Published Online: Nov. 23, 2021

    The Author Email: Ren Jia (jren@zstu.edu.cn)

    DOI:10.3788/AOS202141.2210002

    Topics