English

Utilizing Transfer Learning and a Customized Loss Function for Optic Disc Segmentation from Retinal Images

Image and Video Processing 2020-10-02 v1 Computer Vision and Pattern Recognition

Abstract

Accurate segmentation of the optic disc from a retinal image is vital to extracting retinal features that may be highly correlated with retinal conditions such as glaucoma. In this paper, we propose a deep-learning based approach capable of segmenting the optic disc given a high-precision retinal fundus image. Our approach utilizes a UNET-based model with a VGG16 encoder trained on the ImageNet dataset. This study can be distinguished from other studies in the customization made for the VGG16 model, the diversity of the datasets adopted, the duration of disc segmentation, the loss function utilized, and the number of parameters required to train our model. Our approach was tested on seven publicly available datasets augmented by a dataset from a private clinic that was annotated by two Doctors of Optometry through a web portal built for this purpose. We achieved an accuracy of 99.78\% and a Dice coefficient of 94.73\% for a disc segmentation from a retinal image in 0.03 seconds. The results obtained from comprehensive experiments demonstrate the robustness of our approach to disc segmentation of retinal images obtained from different sources.

Keywords

Cite

@article{arxiv.2010.00583,
  title  = {Utilizing Transfer Learning and a Customized Loss Function for Optic Disc Segmentation from Retinal Images},
  author = {Abdullah Sarhan and Ali Al-KhazÁly and Adam Gorner and Andrew Swift and Jon Rokne and Reda Alhajj and Andrew Crichton},
  journal= {arXiv preprint arXiv:2010.00583},
  year   = {2020}
}

Comments

Accepted by ACCV 2020

R2 v1 2026-06-23T18:56:42.131Z