Skip to content

Commit

Permalink
Update how_to_use_nnunet.md
Browse files Browse the repository at this point in the history
Add how to deploy and run your pretrained model for inference on new images.
  • Loading branch information
ptlzon authored May 30, 2024
1 parent d12a0c1 commit 48e73ad
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions documentation/how_to_use_nnunet.md
Original file line number Diff line number Diff line change
Expand Up @@ -290,6 +290,13 @@ from the respective training). You can pick these files from any of the ensemble
## How to run inference with pretrained models
See [here](run_inference_with_pretrained_models.md)

## How to Deploy and Run Inference with YOUR Pretrained Models
To facilitate the use of pretrained models on a different computer for inference purposes, follow these streamlined steps:
1. Exporting the Model: Utilize the `nnUNetv2_export_model_to_zip` function to package your trained model into a .zip file. This file will contain all necessary model files.
2. Transferring the Model: Transfer the .zip file to the target computer where inference will be performed.
3. Importing the Model: On the new PC, use the `nnUNetv2_install_pretrained_model_from_zip` to load the pretrained model from the .zip file.
Please note that both computers must have nnU-Net installed along with all its dependencies to ensure compatibility and functionality of the model.

[//]: # (## Examples)

[//]: # ()
Expand Down

0 comments on commit 48e73ad

Please sign in to comment.