Add model_seg_rabbit_axon-myelin_bf for download

axondeepseg · Oct 17, 2023 · cf4cfa3 · cf4cfa3
1 parent 4f79a2d
commit cf4cfa3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/download_models.py b/download_models.py
@@ -20,7 +20,7 @@
         'contrasts': ['TEM'],
     },
     'model_seg_rabbit_axon-myelin_bf': {
-        'url': None,
+        'url': 'https://github.com/axondeepseg/model_seg_rabbit_axon-myelin_bf/releases/download/r20231017a/model-nn_seg_rabbit_axon-myelin_bf.zip',
         'description': 'Axon and myelin segmentation on Toluidine Blue stained BF images (rabbit)',
         'contrasts': ['BF'],
     },

diff --git a/nn_axondeepseg.py b/nn_axondeepseg.py
@@ -103,7 +103,7 @@ def main():
         try:
             tmp_dir.mkdir()
         except FileExistsError:
-            delete_dir(tmpr_dir)
+            delete_dir(tmp_dir)
             tmp_dir.mkdir()
         for fname in Path(args.path_dataset).glob('*.png'):
             target_fname = f'{fname.stem}_0000{fname.suffix}'