Skip to content

GitLab

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

d

parent c9f2df7c
Hide whitespace changes
Inline Side-by-side
RobinNet/allModelsCreatePredNpys.py
View file @ 4265dcf2
......@@ -22,26 +22,26 @@ def main():
folds = ["12", "34", "56", "78", "910", "1112", "1314", "1516"]
folder_path = "finalResults/complete_"
unet_models_paths = ["ctfgt2/unet_cv_ctfgt2/",
"seg/unet_seg/",
"seg/unet_cv_seg/",
"thresh/unet_cv_thresh/",
"segmr/mr_unet_cv_segmr/",
"segmr/mr_unet_cv_ctseg/",
"segmr/mr_unet_cv_ctthresh/",
"segmr/mr_unet_cv_unetpred/"]
cluster_models_paths = ["ctfgt2/ae_noclass_cv_ctfgt2/",
"seg/ae_noclass_seg/",
"seg/ae_noclass_cv_seg/",
"thresh/ae_noclass_cv_thresh/"]
class_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/",
"seg/ae_class_seg/",
"seg/ae_class_cv_seg/",
"thresh/ae_class_cv_thresh/"]
clusterclass_models_paths = ["ctfgt2/ae_class_cv_ctfgt2/",
"seg/ae_class_seg/",
"seg/ae_class_cv_seg/",
"thresh/ae_class_cv_thresh/"]
modeltypes = ["Unet_", "Cluster_", "Class_", "Cluster_class_"]
for modeltype in modeltypes:
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 == "Cluster_class_": model_paths = clusterclass_models_paths
......
RobinNet/rerunModels.py
View file @ 4265dcf2
......@@ -19,7 +19,7 @@ from nets.Unet import *
def main():
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 = "Class_"
#modeltype = "Cluster_class_"
......
Markdown is supported
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