fulyvv/EasyMocap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix yolo bug

Browse Source
This commit is contained in:
shuaiqing 2023-07-19 14:00:31 +08:00
parent 0815e2e53d
commit f6ea450543
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
myeasymocap/backbone/yolo/yolo.py
View File

@ -90,7 +90,7 @@ class BaseYOLOv5:
select = np.stack(select) select = np.stack(select)
return select, results return select, results
def select_bbox(self, select, imgname): def select_bbox(self, select, results, imgname):
if select.shape[0] == 0: if select.shape[0] == 0:
return select return select
# Naive: select the best # Naive: select the best