d

RobinNet/13unet_mrt_threshgt.py

@@ -177,15 +177,15 @@ def main():
             #number_patients = number_patients * 5
             img_path = "data/npy_thresh/"
 
-    #specificmodels = [6]
+    specificmodels = [0]
     if cross_val:
         log = open("logs" + modelname + ".txt", "w+")
         log.write(modelname + "\r")
         log.write("Start Cross Validation Training \r")
         log.close()
         print("Start Cross Validation Training")
-        #for validation_round in specificmodels:
-        for validation_round in range(number_patients//2):
+        for validation_round in specificmodels:
+        #for validation_round in range(number_patients//2):
             log = open("logs" + modelname + ".txt", "a+")
             test_patients = (2*validation_round+1, 2*validation_round+2)
             vallist= list(range(1, number_patients+1)) + list(range(1, number_patients+1))