Commit 8cfd1b1f authored by sjromuel's avatar sjromuel
Browse files
parents 18552d7e 4265dcf2
...@@ -22,26 +22,26 @@ def main(): ...@@ -22,26 +22,26 @@ def main():
folds = ["12", "34", "56", "78", "910", "1112", "1314", "1516"] folds = ["12", "34", "56", "78", "910", "1112", "1314", "1516"]
folder_path = "finalResults/complete_" folder_path = "finalResults/complete_"
unet_models_paths = ["ctfgt2/unet_cv_ctfgt2/", unet_models_paths = ["ctfgt2/unet_cv_ctfgt2/",
"seg/unet_seg/", "seg/unet_cv_seg/",
"thresh/unet_cv_thresh/", "thresh/unet_cv_thresh/",
"segmr/mr_unet_cv_ctsegmr/", "segmr/mr_unet_cv_segmr/",
"segmr/mr_unet_cv_ctseg/", "segmr/mr_unet_cv_ctseg/",
"segmr/mr_unet_cv_ctthresh/", "segmr/mr_unet_cv_ctthresh/",
"segmr/mr_unet_cv_unetpred/"] "segmr/mr_unet_cv_unetpred/"]
cluster_models_paths = ["ctfgt2/ae_noclass_cv_ctfgt2/", cluster_models_paths = ["ctfgt2/ae_noclass_cv_ctfgt2/",
"seg/ae_noclass_seg/", "seg/ae_noclass_cv_seg/",
"thresh/ae_noclass_cv_thresh/"] "thresh/ae_noclass_cv_thresh/"]
class_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/", class_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/",
"seg/ae_class_seg/", "seg/ae_class_cv_seg/",
"thresh/ae_class_cv_thresh/"] "thresh/ae_class_cv_thresh/"]
clusterclass_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/", clusterclass_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/",
"seg/ae_class_seg/", "seg/ae_class_cv_seg/",
"thresh/ae_class_cv_thresh/"] "thresh/ae_class_cv_thresh/"]
modeltypes = ["Unet_", "Cluster_", "Class_", "Cluster_class_"] modeltypes = ["Unet_", "Cluster_", "Class_", "Cluster_class_"]
for modeltype in modeltypes: for modeltype in modeltypes:
if modeltype == "Unet_": model_paths = unet_models_paths if modeltype == "Unet_": model_paths = unet_models_paths
elif modeltype == "Cluster_": model_paths == cluster_models_paths elif modeltype == "Cluster_": model_paths = cluster_models_paths
elif modeltype == "Class_": model_paths = class_models_paths elif modeltype == "Class_": model_paths = class_models_paths
elif modeltype == "Cluster_class_": model_paths = clusterclass_models_paths elif modeltype == "Cluster_class_": model_paths = clusterclass_models_paths
...@@ -155,9 +155,6 @@ def main(): ...@@ -155,9 +155,6 @@ def main():
X_img_npys = np.load(img_path + X_img_list[j*3]) X_img_npys = np.load(img_path + X_img_list[j*3])
GT_img_npys = np.load(img_path + GT_img_list[j*3]) GT_img_npys = np.load(img_path + GT_img_list[j*3])
ytrue_img_npys = np.load(img_path + ytrue_img_list[j*3]) ytrue_img_npys = np.load(img_path + ytrue_img_list[j*3])
print(GT_img_list[j*3])
print(GT_img_list[j * 3+1])
print(GT_img_list[j * 3+2])
X_img_npys = np.append(X_img_npys, np.load(img_path + X_img_list[j * 3+1]), axis=0) X_img_npys = np.append(X_img_npys, np.load(img_path + X_img_list[j * 3+1]), axis=0)
GT_img_npys = np.append(GT_img_npys, np.load(img_path + GT_img_list[j * 3+1]), axis=0) GT_img_npys = np.append(GT_img_npys, np.load(img_path + GT_img_list[j * 3+1]), axis=0)
......
...@@ -19,7 +19,7 @@ from nets.Unet import * ...@@ -19,7 +19,7 @@ from nets.Unet import *
def main(): def main():
models = ["12", "34", "56", "78", "910", "1112", "1314", "1516"] models = ["12", "34", "56", "78", "910", "1112", "1314", "1516"]
folder_path = "finalResults/complete_segmr/mr_unet_cv_ctthresh/" folder_path = "finalResults/complete_segmr/mr_unet_cv_unetpred/"
#modeltype = "Cluster_" #modeltype = "Cluster_"
#modeltype = "Class_" #modeltype = "Class_"
#modeltype = "Cluster_class_" #modeltype = "Cluster_class_"
......
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