HTSeg: Hybrid Two-Stage Segmentation Framework for Intestine Segmentation from CT Volumes

This paper proposes a semi-supervised intestine segmentation method from CT volumes. Our method can use densely and sparsely annotated CT volumes for training to reduce the labor of manually annotating intestines. The proposed Hybrid Two-stage Segmentation (HTSeg) framework consists of two networks, a 2D swin-transformer-based network as the first stage and a 3D network as the second stage. In the first stage, we use 6964 labeled CT slices to train the 2D Swin U-Net. The trained 2D Swin U-Net is used to generate pseudo-labels for sparse annotation data. In the second stage, we use sparsely annotated datasets with pseudo-labels and densely annotated datasets to train a 3D multi-view symmetrical network (MVSNet). Experimental results showed that the Dice score of the proposed method was 74.70%, which was 1.03% higher than just using MVSNet. Compared with the other four previous methods (3D U-Net, CPS, EM, MT), the proposed method produced competitive segmentation performance. The code can be found at: https://github.com/MoriLabNU/semi-pseudo-labels.