mirror of
https://github.com/kristoferssolo/Traffic-Light-Detector.git
synced 2025-10-21 20:00:36 +00:00
41 lines
1.0 KiB
Python
Executable File
41 lines
1.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
|
|
from loguru import logger
|
|
from paths import create_dirs, IMAGES_IN_PATH
|
|
|
|
|
|
parser = argparse.ArgumentParser(description="Traffic light detection script.")
|
|
parser.add_argument(
|
|
"-w",
|
|
"--webcam",
|
|
action="store_true",
|
|
help="reads webcam inputs to determine traffic light color",
|
|
)
|
|
parser.add_argument(
|
|
"-i",
|
|
"--image",
|
|
action="store_true",
|
|
help="detects traffic lights in images located in ./assets/images_in/",
|
|
)
|
|
|
|
|
|
@logger.catch
|
|
def main(args) -> None:
|
|
create_dirs()
|
|
if args.webcam:
|
|
from TrafficLightDetector.traffic_light_webcam import TrafficLightDetectorWebcam
|
|
camera = TrafficLightDetectorWebcam(0) # Change number if webcam didn't detect
|
|
camera.enable()
|
|
|
|
if args.image:
|
|
from TrafficLightDetector.traffic_light_images import TrafficLightDetectorImages
|
|
for path in IMAGES_IN_PATH.iterdir():
|
|
image = TrafficLightDetectorImages(path)
|
|
image.draw()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parser.parse_args()
|
|
main(args)
|