Fix entry point for patchnetvlad-download-models

QVPR · Jun 2, 2021 · 35c2ad0 · 35c2ad0
1 parent 2d68577
commit 35c2ad0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -58,7 +58,7 @@
         'console_scripts': ['patchnetvlad-match-two=match_two:main',
                             'patchnetvlad-feature-match=feature_match:main',
                             'patchnetvlad-feature-extract=feature_extract:main',
-                            'patchnetvlad-download-models=download_models:main'],
+                            'patchnetvlad-download-models=download_models:download_all_models'],
       },
       package_data={'': ['configs/*.ini', 'dataset_gt_files/*.npz', 'example_images/*',
                          'output_features/.hidden', 'pretrained_models/.hidden', 'results/.hidden',