Awesome

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