leanderloew / Semi-Supervised-Segmentation-Pytorch

A repository for semi supervised image segmentation using Mean Teacher

Semi-Supervised-Segmentation-Pytorch

A work in progress repository for semi supervised image segmentation using Mean Teacher it includes the following features:

• Easy to train on new Train and Test sets using the provided notebook.
• Different Pre trained Networks.
• Many different Losses.
• TensorboardX integration.
• Hyperparameter tuning.
• Data Loader with Image Augmentation.

TBD: A learned GAN loss. Test on new dataset.

Sources:

• I took the pre trained networks and encoder/decoder code from:
  • https://github.com/CSAILVision/semantic-segmentation-pytorch/blob/master/LICENSE
• Much of the Mean Teacher Code was taken from:
  • https://github.com/CuriousAI/mean-teacher