浏览代码

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,
         }