jpdefrutos commited on
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 + '/' + 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)
 
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)