diff --git a/computer_assisted_intervention/video_seg.ipynb b/computer_assisted_intervention/video_seg.ipynb index 1e18a49cea..f975e06f20 100644 --- a/computer_assisted_intervention/video_seg.ipynb +++ b/computer_assisted_intervention/video_seg.ipynb @@ -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", + "```" ] }, {