fix yolo bug

2023-07-19 14:00:31 +08:00 · 2023-07-19 14:00:31 +08:00 · f6ea450543
commit f6ea450543
parent 0815e2e53d
1 changed files with 1 additions and 1 deletions
--- a/myeasymocap/backbone/yolo/yolo.py
+++ b/myeasymocap/backbone/yolo/yolo.py
@ -90,7 +90,7 @@ class BaseYOLOv5:
        select = np.stack(select)
        return select, results

-    def select_bbox(self, select, imgname):
+    def select_bbox(self, select, results, imgname):
        if select.shape[0] == 0:
            return select
        # Naive: select the best