Skip to content

Commit 02c3486

Browse files
committed
Update readme
Signed-off-by: heyufan <heyufan1995@gmail.com>
1 parent 9236f71 commit 02c3486

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

vista3d/README.md

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,14 @@ pip install -r requirements.txt;
7575
cd ..;
7676
mkdir 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)

vista3d/README_research.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,8 @@ conda create -y -n vista3d python=3.9
5757
conda activate vista3d
5858
pip install -r requirements.txt
5959
mkdir 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
```
6463
The researh repo ussed monai1.3 checkpoint while NVSegment-CTMR used monai1.4, which has subtle layer naming difference.
6564

0 commit comments

Comments
 (0)