Skip to content

add description of loading a pretrained bundle #968

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Nic-Ma merged 5 commits into from
Oct 8, 2022
Merged

add description of loading a pretrained bundle #968

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
5249abc
add description of loading a pretrained bundle
binliunls Sep 30, 2022
8f37157
Merge branch 'Project-MONAI:main' into update-video-seg-with-bundle
binliunls Oct 8, 2022
570241d
change the description of how to download pretrained bundle
binliunls Oct 8, 2022
adf78fc
change the var name from 'pretrained_model' to 'pretrained_weights'
binliunls Oct 8, 2022
e73db34
change the var name from 'pretrained_model' to 'pretrained_weights'
binliunls Oct 8, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion computer_assisted_intervention/video_seg.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -369,7 +369,14 @@
"source": [
"## Normal training loop\n",
"\n",
"A normal training loop will be excuted. Meanwhile, the iou value of each epoch will be recorded as the accuracy."
"A normal training loop will be excuted. Meanwhile, the iou value of each epoch will be recorded as the accuracy. \n",
"There is a pretrained bundle for surgical tool segmentation task in the MONAI model-zoo, the model can be automatically downloaded and loaded to the network by adding codes to the block below:\n",
"```\n",
"pretrained_weights = monai.bundle.load(\n",
" name=\"surgical_tool_segmentation\", bundle_dir=\"./\", version=\"0.2.0\"\n",
")\n",
"model.load_state_dict(pretrained_weights)\n",
"```"
]
},
{
Expand Down