Răsfoiți Sursa

fixed open vocabulary detection res bug

leo-q8 9 luni în urmă
părinte
comite
4d6ff60738

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