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

Added debug file

Browse Source
This commit is contained in:
Kristofers Solo 2022-12-11 17:46:26 +02:00
parent 645c3eabdd
commit 939501896c
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -130,3 +130,4 @@ dmypy.json
assets/* assets/*
model.h5 model.h5
debug

20
main.py
View File

@ -1,8 +1,16 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse import argparse
import importlib import importlib
from pathlib import Path
from detector.paths import create_dirs from detector.paths import BASE_PATH, create_dirs, LOGS_PATH
from loguru import logger
log_level = "DEBUG" if Path.exists(Path.joinpath(BASE_PATH, "debug")) else "INFO"
# Set up logging
logger.add(Path.joinpath(LOGS_PATH, "detection.log"), format="{time} | {level} | {message}", level=log_level, rotation="1 MB", compression="zip")
parser = argparse.ArgumentParser(description="Traffic light detection script.") parser = argparse.ArgumentParser(description="Traffic light detection script.")
@ -30,8 +38,13 @@ parser.add_argument(
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
def main(args) -> None: def main(args) -> None:
create_dirs() create_dirs()
if args.extract: if args.extract:
@ -45,7 +58,10 @@ def main(args) -> None:
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_image() module.detect_traffic_light_color_video()
if args.test:
module = importlib.import_module("detector.test")
module.main()
if __name__ == "__main__": if __name__ == "__main__":