Commit
·
c9f9cef
1
Parent(s):
0978cbc
Updated download URL
Browse files
DeepDeformationMapRegistration/utils/model_downloader.py
CHANGED
@@ -35,7 +35,7 @@ def get_models_path(anatomy: str, model_type: str, output_root_dir: str):
|
|
35 |
assert model_type in MODEL_TYPES.keys(), 'Invalid model type'
|
36 |
anatomy = ANATOMIES[anatomy]
|
37 |
model_type = MODEL_TYPES[model_type]
|
38 |
-
url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained-models/' + anatomy + '
|
39 |
file_path = os.path.join(output_root_dir, 'models', anatomy, model_type + '.h5')
|
40 |
if not os.path.exists(file_path):
|
41 |
os.makedirs(os.path.split(file_path)[0], exist_ok=True)
|
|
|
35 |
assert model_type in MODEL_TYPES.keys(), 'Invalid model type'
|
36 |
anatomy = ANATOMIES[anatomy]
|
37 |
model_type = MODEL_TYPES[model_type]
|
38 |
+
url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained-models/' + anatomy + '_' + model_type + '.h5'
|
39 |
file_path = os.path.join(output_root_dir, 'models', anatomy, model_type + '.h5')
|
40 |
if not os.path.exists(file_path):
|
41 |
os.makedirs(os.path.split(file_path)[0], exist_ok=True)
|