Which model file should i use while deploying custom model that i created using monai label using monai-deploy-app-sdk. #445
Closed
ArpanGyawali
started this conversation in
General
Replies: 1 comment
-
Duplicated |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
My task: Skull stripping
What I did:
Rewrote the config, infers, and trainers file of the segmentation-spleen model within radiology by changing labels, roi, and transforms as required and renamed it as segmentation-brain.
Then, I annotated my own dataset for skull stripping using 3D slicer. The only label used was "brain"
Then I trained the images and its label using the custom segmentation-brain model using monai label extension in 3d slicer and got a dice score of about 0.98 and I am happy with the result.
Now I am trying to deploy the model for inference using monai-deploy-app-sdk.
For that, I used the model.pt file saved inside the apps/radiology/models/segmentation brain directory in the command
python app.py -i input_dir -o output_dir -m model.pt
But I am getting the error: ItemNotExistsError: A predictor of the model is not set
How can I get the saved inference model file that can be used while deploying? Do we need a torchscript file of the model? If yes then how to get that?
Any help would be very appreciated.
Beta Was this translation helpful? Give feedback.
All reactions