Commit
·
674737f
1
Parent(s):
5b0dbe4
Updated download path
Browse files
DeepDeformationMapRegistration/utils/model_utils.py
CHANGED
|
@@ -66,7 +66,7 @@ def load_model(weights_file_path: str, trainable: bool = False, return_registrat
|
|
| 66 |
|
| 67 |
|
| 68 |
def get_spatialtransformer_model():
|
| 69 |
-
url = 'https://github.com/jpdefrutos/DDMR/releases/download/
|
| 70 |
file_path = os.path.join(os.getcwd(), 'models', 'spatialtransformer.h5')
|
| 71 |
if not os.path.exists(file_path):
|
| 72 |
LOGGER.info(f'Model not found. Downloading from {url}... ')
|
|
|
|
| 66 |
|
| 67 |
|
| 68 |
def get_spatialtransformer_model():
|
| 69 |
+
url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained_models_v0/spatialtransformer.h5'
|
| 70 |
file_path = os.path.join(os.getcwd(), 'models', 'spatialtransformer.h5')
|
| 71 |
if not os.path.exists(file_path):
|
| 72 |
LOGGER.info(f'Model not found. Downloading from {url}... ')
|