Utilizing Transfer Learning and a Customized Loss Function for Optic Disc Segmentation from Retinal Images
15th Asian Conference on Computer Vision, ACCV 2020, Virtual, Online, 30 Kasım - 04 Aralık 2020, cilt.12626 LNCS, ss.687-703, (Tam Metin Bildiri)
- Yayın Türü: Bildiri / Tam Metin Bildiri
- Cilt numarası: 12626 LNCS
- Doi Numarası: 10.1007/978-3-030-69541-5_41
- Basıldığı Şehir: Virtual, Online
- Sayfa Sayıları: ss.687-703
- Açık Arşiv Koleksiyonu: AVESİS Açık Erişim Koleksiyonu
- İstanbul Medipol Üniversitesi Adresli: Evet
Özet
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 s. The results obtained from comprehensive experiments demonstrate the robustness of our approach to disc segmentation of retinal images obtained from different sources.