Преглед изворни кода

fixed open vocabulary detection res bug

leo-q8 пре 9 месеци
родитељ
комит
4d6ff60738
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      paddlex/inference/models/open_vocabulary_detection/predictor.py

+ 1 - 1
paddlex/inference/models/open_vocabulary_detection/predictor.py

@@ -117,7 +117,7 @@ class OVDetPredictor(BasicPredictor):
 
         return {
             "input_path": image_paths,
-            "input_img": src_images,
+            "input_img": [img[..., ::-1] for img in src_images],
             "boxes": boxes,
         }