JointRCNN: A Region-Based Convolutional Neural Network for Optic Disc and Cup Segmentation

摘要

Objective: The purpose of this paper is to propose a novel algorithm for joint optic disc and cup segmentation, which aids the glaucoma detection. Methods: By assuming the shapes of cup and disc regions to be elliptical, we proposed an end-to-end region-based convolutional neural network for joint optic disc and cup segmentation (referred to as JointRCNN). Atrous convolution is introduced to boost the performance of feature extraction module. In JointRCNN, disc proposal network (DPN) and cup proposal network (CPN) are proposed to generate bounding box proposals for the optic disc and cup, respectively. Given the prior knowledge that the optic cup is located in the optic disc, disc attention module is proposed to connect DPN and CPN, where a suitable bounding box of the optic disc is first selected and then continued to be propagated forward as the basis for optic cup detection in our proposed network. After obtaining the disc and cup regions, which are the inscribed ellipses of the corresponding detected bounding boxes, the vertical cup-to-disc ratio is computed and used as an indicator for glaucoma detection. Results: Comprehensive experiments clearly show that our JointRCNN model outperforms state-of-the-art methods for optic disc and cup segmentation task and glaucoma detection task. Conclusion: Joint optic disc and cup segmentation, which utilizes the connection between optic disc and cup, could improve the performance of optic disc and cup segmentation. Significance: The proposed method improves the accuracy of glaucoma detection. It is promising to be used for glaucoma screening.