/3dUnet

Primary LanguagePython

3dUnet

  • dataloader_new.py: this only applies to the Huimin's dataset.

  • loss.py:

    • loss for back propagation
    • scores for evaluation (no padding, no background)
  • model_new.py: 3d Unet

  • utils.py: data augmentation

  • visualizer.py: save or load models/curves/predictions

  • train_seq.py: training models

  • test_seq.py: test trained models