Added playing sound option

.gitignore

@@ -130,6 +130,5 @@ dmypy.json
 
 assets/*
 !/assets/haar
-model.h5
 debug
 .logs/*

main.py

@@ -19,10 +19,10 @@ parser.add_argument(
     help="reads webcam inputs to determine traffic light color",
 )
 parser.add_argument(
-    "-c",
+    "-a",
-    "--change",
+    "--audio",
     action="store_true",
-    help="detects traffic lights change",
+    help="plays audio when green light is detected",
 )
 
 
@@ -41,7 +41,7 @@ def main(args) -> None:
         camera = TrafficLightDetectorWebcam(0)  # Change number if webcam didn't detect
         camera.enable()
 
-    if args.change:
+    if args.audio:
         from TrafficLightDetector.traffic_light_webcam import TrafficLightDetectorWebcam
         camera = TrafficLightDetectorWebcam(0, sound=True)  # Change number if webcam didn't detect
         camera.enable()

requirements.txt

@@ -1,2 +1,4 @@
+PyGObject==3.42.2
 loguru==0.6.0
 opencv-python==4.6.0.66
+playsound==1.3.0

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = TrafficLightDetector
-desctiption = Reduce traffic congestion effects due to human reaction times at intersections with traffic lights by utilizing a computer system based on direct communication method of changes in the circumscribed route intersection.
+description = Reduce traffic congestion effects due to human reaction times at intersections with traffic lights by utilizing a computer system based on direct communication method of changes in the circumscribed route intersection.
 author = Kristofers Solo
 license = MIT
 license_file = LICENSE
@@ -11,8 +11,10 @@ classifiers =
 [options]
 packages = TrafficLightDetector
 install_requires =
+    PyGObject>=3.42.2
     loguru>=0.6.0
     opencv-python>=4.6.0.66
+    playsound>=1.3.0
 
 python_requires = >=3.10
 package_dir = =src

src/TrafficLightDetector/traffic_light_detector.py

@@ -22,8 +22,6 @@ class TrafficLightDetector:
     YELLOW = (0, 175, 225)
     GREEN = (0, 150, 0)
 
-    signal_color = ""
-
     def _set_image(self, image=None, roi=None, detectTrafficLights=True) -> None:
         self.image = image
         self.roi = self.image if roi is None else roi
@@ -34,6 +32,7 @@ class TrafficLightDetector:
         self.green = Color("GREEN", self.GREEN, self.GREEN_LOWER, self.GREEN_UPPER, hsv, minDist=30, param2=5)
         self.colors = [self.red, self.yellow, self.green]
         self.detect_traffic_lights = detectTrafficLights
+        self.signal_color = ""
 
     def _outline_traffic_lights(self) -> None:
         gray = cv2.cvtColor(self.image, cv2.COLOR_BGR2GRAY)
@@ -46,22 +45,23 @@ class TrafficLightDetector:
 
     def _draw_circle(self) -> None:
         for color in self.colors:
-            if color.circle is not None:
+            if color.circle is None:
-                for value in color.circle[0, :]:
+                continue
-                    if value[0] > self.size[1] or value[1] > self.size[0] or value[1] > self.size[0] * self.BOUNDARY:
+            for value in color.circle[0, :]:
-                        continue
+                if value[0] > self.size[1] or value[1] > self.size[0] or value[1] > self.size[0] * self.BOUNDARY:
+                    continue
 
-                    h, s = 0, 0
+                h, s = 0, 0
-                    for i in range(-self.RADIUS, self.RADIUS):
+                for i in range(-self.RADIUS, self.RADIUS):
-                        for j in range(-self.RADIUS, self.RADIUS):
+                    for j in range(-self.RADIUS, self.RADIUS):
-                            if (value[1] + i) >= self.size[0] or (value[0] + j) >= self.size[1]:
+                        if (value[1] + i) >= self.size[0] or (value[0] + j) >= self.size[1]:
-                                continue
+                            continue
-                            h += color.mask[value[1] + i, value[0] + j]
+                        h += color.mask[value[1] + i, value[0] + j]
-                            s += 1
+                        s += 1
-                    if h / s > 100:
+                if h / s > 100:
-                        cv2.circle(self.roi if self.detect_traffic_lights else self.image, (value[0], value[1]), value[2] + 10, color.color, 2)  # draws circle
+                    cv2.circle(self.roi if self.detect_traffic_lights else self.image, (value[0], value[1]), value[2] + 10, color.color, 2)  # draws circle
-                        cv2.circle(color.mask, (value[0], value[1]), value[2] + 30, (255, 255, 255), 2)
+                    cv2.circle(color.mask, (value[0], value[1]), value[2] + 30, (255, 255, 255), 2)
-                        if self.TEXT:
+                    if self.TEXT:
-                            cv2.putText(self.roi if self.detect_traffic_lights else self.image, color.name,
+                        cv2.putText(self.roi if self.detect_traffic_lights else self.image, color.name,
-                                        (value[0], value[1]), self.FONT, 1, color.color, 2, cv2.LINE_AA)  # draws text
+                                    (value[0], value[1]), self.FONT, 1, color.color, 2, cv2.LINE_AA)  # draws text
-                        self.signal_color = color.name
+                    self.signal_color = color.name

src/TrafficLightDetector/traffic_light_images.py

@@ -9,12 +9,11 @@ class TrafficLightDetectorImages(TrafficLightDetector):
 
     def __init__(self, path: Path) -> None:
         self.path = path
-        image = cv2.imread(str(path))
+        self._set_image(cv2.imread(str(self.path)))
-        self._set_image(image)
 
     def _save_image(self) -> None:
         cv2.imwrite(str(IMAGES_OUT_PATH.joinpath(self.path.name)), self.image)
 
     def draw(self) -> None:
-        self._draw_circle()
+        self._outline_traffic_lights()
         self._save_image()

src/TrafficLightDetector/traffic_light_webcam.py

@@ -1,6 +1,7 @@
 import cv2
+from paths import SOUND_PATH
+from playsound import playsound
 from TrafficLightDetector.traffic_light_detector import TrafficLightDetector
-from loguru import logger
 
 
 class TrafficLightDetectorWebcam(TrafficLightDetector):
@@ -26,7 +27,5 @@ class TrafficLightDetectorWebcam(TrafficLightDetector):
         cv2.imshow("Video", self.image)
 
     def _make_sound(self) -> None:
-        """Do some sound if green light"""
         if self.signal_color == "GREEN":
-            # DO SOME SOUND
+            playsound(str(SOUND_PATH.joinpath("move.mp3")))
-            logger.debug("DRIVE")

src/paths.py

@@ -10,8 +10,9 @@ ASSETS_PATH = BASE_PATH.joinpath("assets")
 IMAGES_IN_PATH = ASSETS_PATH.joinpath("images_in")
 IMAGES_OUT_PATH = ASSETS_PATH.joinpath("images_out")
 HAAR_PATH = ASSETS_PATH.joinpath("haar").joinpath("TrafficLights.xml")
+SOUND_PATH = ASSETS_PATH.joinpath("sound")
 
-PATHS = (LOGS_PATH, IMAGES_IN_PATH, IMAGES_OUT_PATH)
+PATHS = (LOGS_PATH, IMAGES_IN_PATH, IMAGES_OUT_PATH, SOUND_PATH)
 
 
 log_level = "DEBUG" if BASE_PATH.joinpath("debug").exists() else "INFO"