From f6ea450543752f5ffa9f8e126db8e08df60a7cf4 Mon Sep 17 00:00:00 2001 From: shuaiqing Date: Wed, 19 Jul 2023 14:00:31 +0800 Subject: [PATCH] fix yolo bug --- myeasymocap/backbone/yolo/yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/myeasymocap/backbone/yolo/yolo.py b/myeasymocap/backbone/yolo/yolo.py index 3567e44..74ff54c 100644 --- 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