Skip to content

Commit

Permalink
readme update with link to hugging face weights
Browse files Browse the repository at this point in the history
  • Loading branch information
spirosmaggioros committed Sep 18, 2024
1 parent 0ec20cb commit 4a9ee75
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,15 @@ pip install -e .
```

## Usage
A pre-trained nnUNet model can be found in the [DLICV-V2 v1.0.0 release](https://github.com/CBICA/DLMUSE/releases/tag/v1.0.0).
A pre-trained nnUNet model can be found at our [hugging face account](https://huggingface.co/nichart/DLICV) or at [DLICV-V2 v1.0.0 release](https://github.com/CBICA/DLMUSE/releases/tag/v1.0.0).
Feel free to use it under the package's [licence](LICENCE)
```bash
DLICV -i "input_folder" -o "output_folder" -m "model_path" -f 0 -tr nnUNetTrainer -c 3d_fullres -p nnUNetPlans -d "id" -device cuda/cpu/mps
DLICV -i "input_folder" -o "output_folder" -d "id" -device cuda/cpu/mps
```

You can perform one example test run using the test input folder by running
```bash
DLICV -i test_input/DLICV_test_images -o test_input/DLICV_test_results -m nnunet_results -f 0 -tr nnUNetTrainer -c 3d_fullres -p nnUNetPlans -d 901 -device cuda
DLICV -i test_input/DLICV_test_images -o test_input/DLICV_test_results -d 901 -device cuda
```

## Contact
Expand Down

0 comments on commit 4a9ee75

Please sign in to comment.