final

MIC-DKFZ · May 5, 2024 · 2d2e8ce · 2d2e8ce
1 parent 433dd9c
commit 2d2e8ce
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/nnunetv2/inference/JHU_inference.py b/nnunetv2/inference/JHU_inference.py
@@ -148,8 +148,8 @@ def predict_from_data_iterator(self,
 
 
 if __name__ == '__main__':
-    # python nnunetv2/inference/JHU_inference.py /home/isensee/Downloads/AbdomenAtlasTest /home/isensee/Downloads/AbdomenAtlasTest_pred -model /home/isensee/JHU_trained_model
-    # /home/isensee/JHU_trained_model
+    # python nnunetv2/inference/JHU_inference.py /home/isensee/Downloads/AbdomenAtlasTest /home/isensee/Downloads/AbdomenAtlasTest_pred -model /home/isensee/temp/JHU/trained_model_ep3850
+    # /home/isensee/temp/JHU/trained_model_ep3850
     # /home/isensee/Downloads/AbdomenAtlasTest
     # /home/isensee/Downloads/AbdomenAtlasTest_pred
 
@@ -159,6 +159,7 @@ def predict_from_data_iterator(self,
     parser.add_argument('input_dir', type=str)
     parser.add_argument('output_dir', type=str)
     parser.add_argument('-model', required=True, type=str)
+    parser.add_argument('--disable_tqdm', required=False, action='store_true', default=False)
     args = parser.parse_args()
 
     predictor = JHUPredictor(
@@ -167,9 +168,9 @@ def predict_from_data_iterator(self,
         use_mirroring=True,
         perform_everything_on_device=True,
         device=torch.device('cuda', 0),
-        verbose=True,
+        verbose=False,
         verbose_preprocessing=False,
-        allow_tqdm=True
+        allow_tqdm=not args.disable_tqdm
     )
 
     predictor.initialize_from_trained_model_folder(