File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -75,13 +75,14 @@ pip install -r requirements.txt;
7575cd ..;
7676mkdir NV-Segment-CT/models; mkdir NV-Segment-CTMR/models
7777# download from huggingface link for CT
78- hf download nvidia/NV-Segment-CT vista3d_pretrained_model/model.pt --local-dir NV-Segment-CT/models/ && \
79- mv NV-Segment-CT/models/vista3d_pretrained_model/model.pt NV-Segment-CT/models/model.pt && \
80- rmdir NV-Segment-CT/models/vista3d_pretrained_model
78+ cd NV-Segment-CT
79+ # Option 1: Download using hf and move to expected location
80+ hf download nvidia/NV-Segment-CT --local-dir models/ && \
81+ mv models/vista3d_pretrained_model/model.pt models/model.pt
8182# download from huggingface link for CTMR
82- hf download nvidia /NV-Segment-CTMR vista3d_pretrained_model/model.pt --local-dir NV-Segment-CTMR/models/ && \
83- mv NV-Segment-CTMR/models/vista3d_pretrained_model/model.pt NV-Segment-CTMR/ models/model.pt && \
84- rmdir NV-Segment-CTMR/ models/vista3d_pretrained_model
83+ cd .. /NV-Segment-CTMR
84+ hf download nvidia/ NV-Segment-CTMR --local-dir models/ && \
85+ mv models/vista3d_pretrained_model/model.pt models/model.pt
8586```
8687
8788## 1.1 ** NV-Segment-CT** [[ Github]] ( https://github.com/NVIDIA-Medtech/NV-Segment-CTMR/tree/main/NV-Segment-CT ) [[ Huggingface]] ( https://huggingface.co/nvidia/NV-Segment-CT )
Original file line number Diff line number Diff line change @@ -57,9 +57,8 @@ conda create -y -n vista3d python=3.9
5757conda activate vista3d
5858pip install -r requirements.txt
5959mkdir models
60- hf download nvidia/NV-Segment-CT vista3d_pretrained_model/model_monai1.3.pt --local-dir models/ && \
61- mv models/vista3d_pretrained_model/model_monai1.3.pt models/model.pt && \
62- rmdir models/vista3d_pretrained_model
60+ hf download nvidia/NV-Segment-CT --local-dir models/ && \
61+ mv models/vista3d_pretrained_model/model_monai1.3.pt models/model.pt
6362```
6463The researh repo ussed monai1.3 checkpoint while NVSegment-CTMR used monai1.4, which has subtle layer naming difference.
6564
You can’t perform that action at this time.
0 commit comments