setup.py 626 B

12345678910111213141516171819
  1. import argparse
  2. from detection.detection_setup import setup
  3. def main():
  4. ap = argparse.ArgumentParser()
  5. ap.add_argument("-i", "--input", required=True, help="Uses this input as image for setup")
  6. ap.add_argument("-s", "--scale", type=float, default=0.25, help="Scales image by this factor (default: x0.25)")
  7. ap.add_argument("-c", "--config", type=str, default="detection/config.json",
  8. help="Saves config under this filename (default: detection/config.json)")
  9. args = vars(ap.parse_args())
  10. setup(args['input'], args['config'], args['scale'])
  11. if __name__ == "__main__":
  12. main()