Joint disc and cup segmentation based on recurrent fully convolutional network

PLoS One. 2020 Sep 21;15(9):e0238983. doi: 10.1371/journal.pone.0238983. eCollection 2020.

Abstract

The optic disc(OD) and the optic cup(OC) segmentation is an key step in fundus medical image analysis. Previously, FCN-based methods have been proposed for medical image segmentation tasks. However, the consecutive convolution and pooling operations usually hinder dense prediction tasks which require detailed spatial information, such as image segmentation. In this paper, we propose a network called Recurrent Fully Convolution Network(RFC-Net) for automatic joint segmentation of the OD and the OC, which can captures more high-level information and subtle edge information. The RFC-Net can minimize the loss of spatial information. It is mainly composed of multi-scale input layer, recurrent fully convolutional network, multiple output layer and polar transformation. In RFC-Net, the multi-scale input layer constructs an image pyramid. We propose four recurrent units, which are respectively applied to RFC-Net. Recurrent convolution layer effectively ensures feature representation for OD and OC segmentation tasks through feature accumulation. For each multiple output image, the multiple output cross entropy loss function is applied. To better balance the cup ratio of the segmented image, the polar transformation is used to transform the fundus image from the cartesian coordinate system to the polar coordinate system. We evaluate the effectiveness and generalization of the proposed method on the DRISHTI-GS1 dataset. Compared with the original FCN method and other state-of-the-art methods, the proposed method achieves better segmentation performance.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Databases, Genetic
  • Eye / diagnostic imaging
  • Eye / embryology
  • Fundus Oculi
  • Glaucoma
  • Humans
  • Image Processing, Computer-Assisted / methods*
  • Meniscus
  • Neural Networks, Computer
  • Optic Disk / diagnostic imaging*
  • Optical Imaging / methods*

Grants and funding

This work was supported by the National Natural Science Foundation of China under Grant 61962054.