FundusPosNet: A Deep Learning Driven Heatmap Regression Model for the Joint Localization of Optic Disc and Fovea Centers in Color Fundus Images

Bhatkalkar, Bhargav J and Nayak, Vignesha and Shenoy, Sathvik V and Arjunan, Vijaya R (2021) FundusPosNet: A Deep Learning Driven Heatmap Regression Model for the Joint Localization of Optic Disc and Fovea Centers in Color Fundus Images. IEEE Access, 9. ISSN 2169-3536

[img] PDF
13976.pdf - Published Version
Restricted to Registered users only

Download (1MB) | Request a copy

Abstract

The localization of the optic disc and fovea is crucial in the automated diagnosis of vari�ous retinal diseases. We propose a novel deep learning driven heatmap regression model based on the encoder-decoder architecture for the joint detection of optic disc and fovea centers in color fundus images. To train the regression model, we transform the ground-truth center coordinates of optic disc and fovea of the IDRiD dataset to heatmaps using a 2D-Gaussian equation. The model is capable of pinpointing any single pixel in a vast 2D image space. The model is tested on IDRiD test dataset, Messidor, and G1020 datasets. The model outperforms the state-of-the-art methods on these datasets. The model is very robust and generic, which can be trained and used for the simultaneous localization of multiple landmarks in different medical image datasets. The full implementation code and the trained model with weights (based on Keras) are available for reuse at https://github.com/bhargav-jb/FundusPosNet.

Item Type: Article
Uncontrolled Keywords: Fundus image, optic disc, fovea, deep learning, heatmap, regression neural network, Gaussian blob
Subjects: Engineering > MIT Manipal > Computer Science and Engineering
Depositing User: MIT Library
Date Deposited: 03 Mar 2022 08:37
Last Modified: 03 Mar 2022 08:37
URI: http://eprints.manipal.edu/id/eprint/158360

Actions (login required)

View Item View Item