Downloading the weights of the model

[tomDag25]

Hey, I was wondering how I could just download the weights of totalspineseg ? I want to use the model offline and I need to be able to access the model without internet access.
Thanks !

[yw7]

Hi @tomDag25,

Thank you for using TotalSpineSeg!

To install the model weights offline, please follow these steps:

1. Install TotalSpineSeg:

   • Follow the installation instructions provided in the README.

2. Download the model weights from the following links:

   • Dataset101_TotalSpineSeg_step1_r20241115.zip
   • Dataset102_TotalSpineSeg_step2_r20241115.zip

3. Determine the extraction directory:

   • By default, extract the weights to: totalspineseg/models/nnUNet/results
   • If you've set the TOTALSPINESEG_DATA environment variable, extract the weights to: $TOTALSPINESEG_DATA/nnUNet/results

If you have further questions or need additional assistance, please don't hesitate to ask.

Additionally, please note that TotalSpineSeg has been integrated into the Spinal Cord Toolbox, providing an alternative platform for utilization: https://spinalcordtoolbox.com/.