Recurrent U-Net for Resource-Constrained Segmentation

Wei Wang,Kaicheng Yu,Joachim Hugonot,Pascal Fua,Mathieu Salzmann
DOI: https://doi.org/10.48550/arXiv.1906.04913
2019-06-12
Abstract:State-of-the-art segmentation methods rely on very deep networks that are not always easy to train without very large training datasets and tend to be relatively slow to run on standard GPUs. In this paper, we introduce a novel recurrent U-Net architecture that preserves the compactness of the original U-Net, while substantially increasing its performance to the point where it outperforms the state of the art on several benchmarks. We will demonstrate its effectiveness for several tasks, including hand segmentation, retina vessel segmentation, and road segmentation. We also introduce a large-scale dataset for hand segmentation.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?