Liver and Tumor Segmentation for CT Images with U-Net

Liver and Tumor Segmentation for CT Images with U-Net

  • This is the final project for MIT 6.869 Spring 2022.
  • The source code is available on my github repo.

Method

  • Network structure: U-Net with 16 features on the first layer. Optimizer was Adam. Loss function was crossEntropy
  • Data set: liver data set from Medical Segmentation Decathlon. I sliced the 3-D CT images into 3-channel 2-D images for training and validation.
  • An experimental idea of fusing the prediction labels along the three directions.

Result

CT image True label Prediction Fused prediction
Share