Browse Source

fixed open vocabulary detection res bug

leo-q8 9 months ago
parent
commit
4d6ff60738
1 changed files with 1 additions and 1 deletions
  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,
         }