Fix color bug

utils/__init__.py

@@ -84,7 +84,7 @@ class Predictor:
       cv2.putText(image, text, (x1, y1 - 5), 
                     cv2.FONT_HERSHEY_DUPLEX, 0.3, (255, 255, 255), 1)
 
-    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    return image
 
   def annotate_masks(self, image, cls, confs, masks, results):
     # image, cls, confs, _, masks, results = self.predict(image_path)
@@ -114,7 +114,7 @@ class Predictor:
       cv2.putText(image, text, (cx - int(0.5 * w), cy),
                 cv2.FONT_HERSHEY_DUPLEX, 0.5, (255, 255, 255), 1)
       
-    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    return image
   
   def transform(self, image_path, annot_boxes=False, annot_masks=False):
       image, cls, confs, boxes, masks, results = self.predict(image_path)