From 2eee620844e0b67300d9a6226405012fe20a3687 Mon Sep 17 00:00:00 2001 From: Fabian Isensee Date: Tue, 30 Apr 2024 20:52:34 +0200 Subject: [PATCH] update model --- nnunetv2/inference/JHU_inference.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/nnunetv2/inference/JHU_inference.py b/nnunetv2/inference/JHU_inference.py index c5a59ab40..1abcb7d26 100644 --- a/nnunetv2/inference/JHU_inference.py +++ b/nnunetv2/inference/JHU_inference.py @@ -148,7 +148,7 @@ def predict_from_data_iterator(self, if __name__ == '__main__': - # /home/isensee/Downloads/jhu_preliminary_trained_model/nnUNetTrainer_fasterDA_minVal_float16data__nnUNetResEncUNetLPlans_torchres__3d_fullres + # /home/isensee/JHU_trained_model # /home/isensee/Downloads/AbdomenAtlasTest # /home/isensee/Downloads/AbdomenAtlasTest_pred @@ -174,7 +174,7 @@ def predict_from_data_iterator(self, predictor.initialize_from_trained_model_folder( args.model, ('all', ), - 'checkpoint_anon.pth' # TODO update + 'checkpoint_latest.pth' ) # we need to create list of list of input files @@ -187,8 +187,8 @@ def predict_from_data_iterator(self, output_folders, save_probabilities=False, overwrite=True, - num_processes_preprocessing=3, - num_processes_segmentation_export=6, + num_processes_preprocessing=2, + num_processes_segmentation_export=3, folder_with_segs_from_prev_stage=None, num_parts=1, part_id=0