Experimental Science
Retinal image quality assessment based on FA-Net
Cheng Wan, Qijing You, Jing Sun, Jianxin Shen, Qiuli Yu
Published 2019-08-10
Cite as Chin J Exp Ophthalmol, 2019, 37(8): 608-612. DOI: 10.3760/cma.j.issn.2095-0160.2019.08.004
Abstract
ObjectiveTo propose a deep learning-based retinal image quality classification network, FA-Net, to make convolutional neural network (CNN) more suitable for image quality assessment in eye disease screening system.
MethodsThe main network of FA-Net was composed of VGG-19.On this basis, attention mechanism was added to the CNN.By using transfer learning method in training, the weight of ImageNet was used to initialize the network.The attention net is based on foreground extraction by extracting the blood vessel and suspected regions of lesion and assigning higher weights to region of interest to enhance the learning of these important areas.
ResultsTotal of 2 894 fundus images were used for training FA-Net.FA-Net achieved 97.65% classification accuracy on a test set containing 2 170 fundus images, with the sensitivity and specificity of 0.978 and 0.960, respectively, and the area under curve(AUC) was 0.995.
ConclusionsCompared with other CNNs, the proposed FA-Net has better classification performance and can evaluate retinal fundus image quality more accurately and efficiently.The network takes into account the human visual system (HVS) and human attention mechanism.By adding attention module into the VGG-19 network structure, the classification results can be better interpreted as well as better classification performance.
Key words:
Retinal image quality assessment; Retinal image enhancement; Convolutional neural networks; Deep learning
Contributor Information
Cheng Wan
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Qijing You
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Jing Sun
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Jianxin Shen
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Qiuli Yu
Department of Ophthalmology, BenQ Hospital of Nanjing Medical University, Nanjing 210019, China