osbm commited on
Commit
1a5ed5f
1 Parent(s): 262cadf

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -8
main.py CHANGED
@@ -100,18 +100,18 @@ class csPCaAlgorithm(SegmentationAlgorithm):
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("/opt/algorithm/nnunet/input")
113
- self.nnunet_out_dir = Path("/opt/algorithm/nnunet/output")
114
- self.nnunet_results = Path("/opt/algorithm/results")
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)