From 58816073736bc794d4a5aca980551f4e7adc04fa Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Sun, 11 Dec 2022 17:53:15 +0200 Subject: [PATCH] Disabled video support --- main.py | 25 +++++++++---------------- 1 file changed, 9 insertions(+), 16 deletions(-) diff --git a/main.py b/main.py index 3ef02a0..500cf5f 100755 --- a/main.py +++ b/main.py @@ -32,16 +32,12 @@ parser.add_argument( action="store_true", help="detects traffic lights in images located in ./assets/detection/images_in/", ) -parser.add_argument( - "-v", - "--video", - action="store_true", - help="detects traffic lights in videos located in ./assets/detection/videos_in/", -) -parser.add_argument( - "--test", - action="store_true", -) +# parser.add_argument( +# "-v", +# "--video", +# action="store_true", +# help="detects traffic lights in videos located in ./assets/detection/videos_in/", +# ) @logger.catch @@ -56,12 +52,9 @@ def main(args) -> None: if args.image: module = importlib.import_module("detector.detect_traffic_light_color_image") module.detect_traffic_light_color_image() - if args.video: - module = importlib.import_module("detector.detect_traffic_light_color_video") - module.detect_traffic_light_color_video() - if args.test: - module = importlib.import_module("detector.test") - module.main() + # if args.video: + # module = importlib.import_module("detector.detect_traffic_light_color_video") + # module.detect_traffic_light_color_video() if __name__ == "__main__":