stitchnet

Weights for Unet and StitchNet can be found at here. The data folder, cleaned_cropped_full_ct, does not contain any actual data. The data must be downloaded from here. Once downloaded, you can follow the pre-proccessing procedure outlined in "Semi-Supervised COVID-19 CT Image Segmentation Using Deep Generative Models".

Dependencies

  • tensorflow 2.7.0
  • albumentations 1.1.0
  • segmentation-models 1.0.1
  • tensorflow-probability 0.15.0
  • focal-loss 0.0.7