Update utils/object_detection_brainai.py

utils/object_detection_brainai.py

@@ -17,12 +17,7 @@ class ObjectDetectionModel():
         with open("utils/game_classes.txt", "r") as file:
            self.game_classes = [line.strip() for line in file if line.strip()]
 
-    def process(self, img):
-        result = self.model(img)
-        img_plot = result[0].plot()
-
-        return img_plot
-
+    
     def process_image(self, img):
 
         if isinstance(img, np.ndarray):
@@ -30,6 +25,8 @@ class ObjectDetectionModel():
         else:
             uploaded_img = PIL.Image.open(img)
             uploaded_img_cv =  np.array(uploaded_img) 
+            if uploaded_img_cv.shape[-1] == 4:  
+                uploaded_img_cv = cv2.cvtColor(uploaded_img_cv, cv2.COLOR_RGBA2RGB)  
 
         result = self.model(uploaded_img_cv, verbose=False)
         img_plot = result[0].plot()