| 80 | |
| 81 | |
| 82 | def setup_object_detection(): |
| 83 | call(PULL_OBJECT_DETECTION, shell=True) |
| 84 | setup( |
| 85 | name='object_detection', |
| 86 | version='0.1', |
| 87 | include_package_data=True, |
| 88 | packages=[p for p in find_packages() if p.startswith('object_detection')], |
| 89 | description='Tensorflow Object Detection Library', |
| 90 | cmdclass={ |
| 91 | args.type: InstallCommand, |
| 92 | } |
| 93 | ) |
| 94 | setup( |
| 95 | name='slim', |
| 96 | version='0.1', |
| 97 | include_package_data=True, |
| 98 | packages=[p for p in find_packages() if p.startswith('slim')], |
| 99 | description='tf-slim', |
| 100 | cmdclass={ |
| 101 | args.type: InstallCommand, |
| 102 | } |
| 103 | ) |
| 104 | shutil.rmtree('slim.egg-info') |
| 105 | shutil.rmtree('object_detection.egg-info') |
| 106 | |
| 107 | |
| 108 | def setup_classification(): |