Deep level set learning for optic disc and cup segmentation
| Autor: | Jinhui Zhu, Huichou Huang, Jiang Liu, Qingyao Wu, Pengshuai Yin, Yanwu Xu, Chang'an Yi |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
Artificial neural network
business.industry Computer science Cognitive Neuroscience Deep learning Boundary (topology) Image segmentation Computer Science Applications Term (time) Level set medicine.anatomical_structure Artificial Intelligence medicine Computer vision Segmentation Artificial intelligence business Optic disc |
| Zdroj: | Neurocomputing. 464:330-341 |
| ISSN: | 0925-2312 |
| Popis: | Optic disc and cup segmentation play an essential step towards automatic retinal diagnose system. The task is very challenging since the boundary between optic disc and cup is weak and the existing segmentation network with cross-entropy loss is hard to inject domain-specific knowledge. To solve the problem, we propose a level set based deep learning method for optic disc and cup segmentation. Particularly, we treat the output of the neural network as a level set and add several constraints to make the predicted level set satisfy some characteristics, such as the length constraint and region constraint. The length term lets the boundary tend to smooth while the region term lets the response inside the predicted area tend to be the same. The region term considers the relationship between pixels inside optic disc or cup while the cross-entropy loss treats the segmentation as a pixel-wise classification without considering the relationship between pixels. We conduct extensive experiments on several datasets including ORIGA and REFUGE and DRISHTI-GS dataset. The experiment results verify the effectiveness of our method. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|
Full Text from ScienceDirect