fix rgb to bgr

app.py

@@ -14,7 +14,8 @@ with open("classes.names", 'r') as f:
 
 
 def predict(inp):
-    img = cv2.resize(inp, (700, 700))
+    img_bgr = inp.astype('uint8')[...,::-1]
+    img = cv2.resize(img_bgr, (700, 700))
     height, width, channels = img.shape
 
     # Convert image into blob and load it on model
@@ -65,7 +66,10 @@ def predict(inp):
         cv2.putText(img, label + " " + conf_value, (x, y-10),
                     cv2.FONT_HERSHEY_COMPLEX, 1, (0, 255, 0), 2)
     # return the images
-    return img
+    cv2.imwrite("out.jpg", img)
+
+    img_rgb = img[...,::-1]
+    return img_rgb
 
 
 gr.Interface(

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "yolo-volley-detection"
+version = "0.1.0"
+description = ""
+authors = ["nicola.landro <[email protected]>"]
+readme = "README.md"
+packages = [{include = "yolo_volley_detection"}]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+matplotlib = "^3.6.2"
+gradio = "^3.9.1"
+opencv-python = "^4.6.0.66"
+numpy = "^1.23.4"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"