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", action="store_true",
help="detects traffic lights in images located in ./assets/detection/images_in/", help="detects traffic lights in images located in ./assets/detection/images_in/",
) )
parser.add_argument( # parser.add_argument(
"-v", # "-v",
"--video", # "--video",
action="store_true", # action="store_true",
help="detects traffic lights in videos located in ./assets/detection/videos_in/", # help="detects traffic lights in videos located in ./assets/detection/videos_in/",
) # )
parser.add_argument(
"--test",
action="store_true",
)
@logger.catch @logger.catch
@ -56,12 +52,9 @@ def main(args) -> None:
if args.image: if args.image:
module = importlib.import_module("detector.detect_traffic_light_color_image") module = importlib.import_module("detector.detect_traffic_light_color_image")
module.detect_traffic_light_color_image() module.detect_traffic_light_color_image()
if args.video: # if args.video:
module = importlib.import_module("detector.detect_traffic_light_color_video") # module = importlib.import_module("detector.detect_traffic_light_color_video")
module.detect_traffic_light_color_video() # module.detect_traffic_light_color_video()
if args.test:
module = importlib.import_module("detector.test")
module.main()
if __name__ == "__main__": if __name__ == "__main__":