Commit 41cd9670 authored by sjromuel's avatar sjromuel
Browse files

g

parent 59bbe025
......@@ -60,12 +60,12 @@ def main():
print('GT Type:', gt_type)
### Load test patient
if gt_type == "thresh":
if gt_type == "thresh" or gt_type == "ctthresh_gt":
img_path = "data/npy_thresh/"
else:
img_path = "data/npy/"
full_list = os.listdir(img_path)
seg_list = os.listdir("../data/npy/")
seg_list = os.listdir("data/npy/")
X_img_list = []
GT_img_list = []
ytrue_img_list = []
......
......@@ -61,7 +61,7 @@ def main():
print('GT Type:', gt_type)
### Load test patient
if gt_type == "thresh":
if gt_type == "thresh" or gt_type == "ctthresh_gt":
img_path = "data/npy_thresh/"
else:
img_path = "data/npy/"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment