fix: not loading default checkpoint

Former-commit-id: 5888ce081945fda5203daa53c874281ac2ce64d1

object-locator/locate.py

@@ -84,7 +84,7 @@
     # Pretrained models that come with this package
     if args.model == 'unet_256x256_sorghum':
         args.model = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                                  'models',
+                                  'checkpoints',
                                   'unet_256x256_sorghum.ckpt')
     if os.path.isfile(args.model):
         if args.cuda:

setup.py

@@ -6,7 +6,7 @@
     description='Object Location using PyTorch (inference/testing only).',
 
     # The project's main homepage.
-    url='https://viperlab.org',
+    url='https://engineering.purdue.edu/~sorghum',
 
     # Author details
     author='Javier Ribera, Yuhao Chen, and Edward Delp',
@@ -25,19 +25,22 @@
     ],
     python_requires='~=3.6',
     # What does your project relate to?
-    keywords='object localization purdue',
+    keywords='object localization location purdue',
 
     # You can just specify the packages manually here if your project is
     # simple. Or you can use find_packages().
-    packages=find_packages(),
+    packages=['object-locator', 'object-locator.models'],
+    package_dir={'object-locator': 'object-locator'},
 
-    package_data={'object-locator': ['models/*.ckpt']},
+    package_data={'object-locator': ['checkpoints/*.ckpt']},
+    include_package_data=True,
 
     # List run-time dependencies here.  These will be installed by pip when
     # your project is installed. For an analysis of "install_requires" vs pip's
     # requirements files see:
     # https://packaging.python.org/en/latest/requirements.html
-    install_requires=['matplotlib', 'numpy',
-                      'scikit-image', 'tqdm', 'argparse', 'parse',
-                      'scikit-learn', 'pandas'],
+    # (We actually use conda for dependency management)
+    # install_requires=['matplotlib', 'numpy',
+    #                   'scikit-image', 'tqdm', 'argparse', 'parse',
+    #                   'scikit-learn', 'pandas'],
 )