chore: added green and refactor

inference/colors.py

@@ -28,6 +28,12 @@
     "upper_hsv": (29, 255, 255),
 }
 
+green = {
+    "name": "green",
+    "lower_hsv": (48, 50, 0),
+    "upper_hsv": (55, 255, 255),
+}
+
 sky_blue = {
     "name": "sky_blue",
     "lower_hsv": (95, 38, 0),
@@ -43,7 +49,7 @@
 black = {
     "name": "black",
     "lower_hsv": (0, 0, 0),
-    "upper_hsv": (179, 255, 60),
+    "upper_hsv": (179, 255, 49),
 }
 
-all = [white, red, blueish_red, orange, yellow, sky_blue, blue, black]
+all = [white, red, orange, yellow, green, sky_blue, blue, blueish_red, black]

inference/filters.py

@@ -1,8 +1,8 @@
-from inference.colors import black, blue, sky_blue, yellow
+from inference.colors import black, blue, green, sky_blue
 
 chelsea_filter = {
     "name": "Chelsea",
-    "colors": [blue, yellow],
+    "colors": [blue, green],
 }
 
 city_filter = {

inference/hsv_classifier.py

@@ -1,6 +1,5 @@
 import copy
-import json
-from typing import List, Tuple
+from typing import List
 
 import cv2
 import matplotlib.pyplot as plt
@@ -20,14 +19,6 @@ def __init__(self, filters: List[dict]):
         filters: List[dict]
             List of colors to classify
 
-            If you want to add a specific color, you can add it as a Python dictionary with the following format:
-
-            custom_color = {
-                "name":"my_custom_color",
-                "lower_hsv": (0, 0, 0),
-                "upper_hsv": (179, 255, 255)
-            }
-
             Format:
             [
                 {
@@ -48,6 +39,15 @@ def __init__(self, filters: List[dict]):
                 },
             ]
 
+            If you want to add a specific color, you can add it as a Python dictionary with the following format:
+
+            custom_color = {
+                "name":"my_custom_color",
+                "lower_hsv": (0, 0, 0),
+                "upper_hsv": (179, 255, 255)
+            }
+
+            You can find your custom hsv range with an online tool like https://github.com/hariangr/HsvRangeTool
         """
         super().__init__()
 

soccer/draw.py

@@ -609,6 +609,9 @@ def draw_arrow_head(
         # vector length
         Len = sqrt(dX * dX + dY * dY)  # use Hypot if available
 
+        if Len == 0:
+            return img
+
         # normalized direction vector components
         udX = dX / Len
         udY = dY / Len