Commit 4265dcf2 authored by Robin's avatar Robin
Browse files

d

parent c9f2df7c
...@@ -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_segmr/", "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
......
...@@ -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