veb-101/UWMGI_Medical_Image_Segmentation

Medical Image Segmentation Gradio App

For the Gradio app we've removed the dependency on pytorch-lightning otherwise used in the project. The load_lightning_SD_to_Usual_SD.ipynb notebook contains the steps used to convert pytorch-lightning checkpoint to a regular model checkpoint. This was mainly done to reduce the file size (977 MB --> 244 MB).

You can download the original saved checkpoint from over here: wandb artifact

Or via Python:

import wandb
run = wandb.init()
artifact = run.use_artifact('veb-101/UM_medical_segmentation/model-jsr2fn8v:v0', type='model')
artifact_dir = artifact.download()