CFUN: Combining Faster R-CNN and U-net Network for Efficient Whole Heart Segmentation

[guanfuchen]

related paper

摘要

In this paper,we propose a novel heart segmentation pipeline Combining Faster R-CNN and U-net Network (CFUN). Due to Faster R-CNN’s precise localization ability and U-net’s powerful segmentation ability, CFUN needs only one-step detection and segmentation inference to get the whole heart segmentation result, obtaining good results with significantly reduced computational cost. Besides, CFUN adopts a new loss function based on edge information named 3D Edge-loss as an auxiliary loss to accelerate the convergence of training and improve the segmentation results. Extensive experiments on the public dataset show that CFUN exhibits competitive segmentation performance in a sharply reduced inference time. Our source code and the model are publicly available at https://github.com/Wuziyi616/CFUN.

---

概述

组合Faster R-CNN（定位能力）和U-Net（分割能力）用于心脏分割，提出了3D Edge-loss作为辅助损失加速收敛提升分割结果。