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".
- tensorflow 2.7.0
- albumentations 1.1.0
- segmentation-models 1.0.1
- tensorflow-probability 0.15.0
- focal-loss 0.0.7