Skip to content

Commit 0f21835

Browse files
binliunlspre-commit-ci[bot]KumoLiu
authored
Hugging face pipeline (#1946)
Fixes #1945 . ### Description Add a tutorial to show how to rewrite a spleen ct segmentation task into a hugging face pipeline so that users can easily call and distribute it. ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [ ] Avoid including large-size files in the PR. - [ ] Clean up long text outputs from code cells in the notebook. - [ ] For security purposes, please check the contents and remove any sensitive info such as user names and private key. - [ ] Ensure (1) hyperlinks and markdown anchors are working (2) use relative paths for tutorial repo files (3) put figure and graphs in the `./figure` folder - [ ] Notebook runs automatically `./runner.sh -t <path to .ipynb file>` --------- Signed-off-by: binliu <binliu@nvidia.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
1 parent 7669d45 commit 0f21835

File tree

7 files changed

+1030
-0
lines changed

7 files changed

+1030
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,12 @@ This tutorial shows how to train a surgical tool segmentation model to locate to
226226
##### [endoscopic inbody classification](./computer_assisted_intervention/endoscopic_inbody_classification.ipynb)
227227
Tutorial to show the pipeline of fine tuning an endoscopic inbody classification model based on a corresponding pretrained bundle.
228228

229+
#### <ins>**Hugging Face**</ins>
230+
##### [MONAI Hugging Face Pipeline](./hugging_face/hugging_face_pipeline_for_monai.ipynb)
231+
This tutorial demonstrates how to encapsulate an existing MONAI model workflow into a Hugging Face pipeline, which is widely adopted by the open-source community.
232+
##### [Fine-tuning for Hugging Face Pipeline](./hugging_face/finetune_vista3d_for_hugging_face_pipeline.ipynb)
233+
This tutorial explains the process of fine-tuning a VISTA3D model and integrating it into a Hugging Face pipeline for inference.
234+
229235
#### <ins>**Modules**</ins>
230236
##### [bundle](./bundle)
231237
Get started tutorial and concrete training / inference examples for MONAI bundle features.

figures/hugging_face_general.jpg

190 KB
Loading

figures/hugging_face_model.jpg

144 KB
Loading

figures/hugging_face_pipeline.jpg

109 KB
Loading

hugging_face/finetune_vista3d_for_hugging_face_pipeline.ipynb

Lines changed: 426 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)