Update main.py

main.py

@@ -246,6 +246,12 @@ class ProstateSegmentationAlgorithm(SegmentationAlgorithm):
 def predict(input_file):
     print("Making prediction")
     image = sitk.ReadImage(input_file)
+    
+    os.makedirs("./input/images/transverse-t2-prostate-mri/", exist_ok=True)
+    os.makedirs("./output/images/softmax-prostate-peripheral-zone-segmentation", exist_ok=True)
+    os.makedirs("./output/images/softmax-prostate-central-gland-segmentation", exist_ok=True)
+    os.makedirs("./output/images/prostate-zonal-segmentation", exist_ok=True)
+    
     sitk.WriteImage(image, "./input/images/transverse-t2-prostate-mri/1009_2222_t2w.mha")
     
     ProstateSegmentationAlgorithm().process()