Bläddra i källkod

Fix generic object detection postprocess

Bobholamovic 2 år sedan
förälder
incheckning
b4c0f12ab2
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      paddlers/tasks/object_detector.py

+ 1 - 1
paddlers/tasks/object_detector.py

@@ -771,7 +771,7 @@ class BaseDetector(BaseModel):
                     dt = bboxes[k]
                     dt = bboxes[k]
                     k = k + 1
                     k = k + 1
                     dt = dt.tolist()
                     dt = dt.tolist()
-                    if len(dt) == 8:
+                    if len(dt) == 6:
                         # Generic object detection
                         # Generic object detection
                         num_id, score, xmin, ymin, xmax, ymax = dt
                         num_id, score, xmin, ymin, xmax, ymax = dt
                         w = xmax - xmin
                         w = xmax - xmin