kristoferssolo/Traffic-Light-Detector
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Traffic-Light-Detector.git synced 2025-10-21 20:00:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Disabled video support

Browse Source
This commit is contained in:
Kristofers Solo 2022-12-11 17:53:15 +02:00
parent 3592fdb142
commit 5881607373
1 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
main.py
View File

@ -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__":