Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -100,18 +100,18 @@ class csPCaAlgorithm(SegmentationAlgorithm):
|
|
100 |
|
101 |
# input / output paths for algorithm
|
102 |
self.image_input_dirs = [
|
103 |
-
"
|
104 |
-
"
|
105 |
-
"
|
106 |
]
|
107 |
self.scan_paths = []
|
108 |
-
self.cspca_detection_map_path = Path("
|
109 |
-
self.case_confidence_path = Path("
|
110 |
|
111 |
# input / output paths for nnUNet
|
112 |
-
self.nnunet_inp_dir = Path("
|
113 |
-
self.nnunet_out_dir = Path("
|
114 |
-
self.nnunet_results = Path("
|
115 |
|
116 |
# ensure required folders exist
|
117 |
self.nnunet_inp_dir.mkdir(exist_ok=True, parents=True)
|
|
|
100 |
|
101 |
# input / output paths for algorithm
|
102 |
self.image_input_dirs = [
|
103 |
+
"./input/images/transverse-t2-prostate-mri",
|
104 |
+
"./input/images/transverse-adc-prostate-mri",
|
105 |
+
"./input/images/transverse-hbv-prostate-mri",
|
106 |
]
|
107 |
self.scan_paths = []
|
108 |
+
self.cspca_detection_map_path = Path("./output/images/cspca-detection-map/cspca_detection_map.mha")
|
109 |
+
self.case_confidence_path = Path("./output/cspca-case-level-likelihood.json")
|
110 |
|
111 |
# input / output paths for nnUNet
|
112 |
+
self.nnunet_inp_dir = Path("./nnunet/input")
|
113 |
+
self.nnunet_out_dir = Path("./nnunet/output")
|
114 |
+
self.nnunet_results = Path("./results")
|
115 |
|
116 |
# ensure required folders exist
|
117 |
self.nnunet_inp_dir.mkdir(exist_ok=True, parents=True)
|