mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix batch inference bug (#413)

pull/415/head
RangiLyu 2022-12-29 17:18:54 +08:00 committed by GitHub
parent 99af3c980f
commit cdc359c2de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmyolo/models/dense_heads/yolov5_head.py
View File

@ -365,7 +365,7 @@ class YOLOv5Head(BaseDenseHead):
] ]
flatten_objectness = torch.cat(flatten_objectness, dim=1).sigmoid() flatten_objectness = torch.cat(flatten_objectness, dim=1).sigmoid()
else: else:
flatten_objectness = [None for _ in range(len(featmap_sizes))] flatten_objectness = [None for _ in range(num_imgs)]
results_list = [] results_list = []
for (bboxes, scores, objectness, for (bboxes, scores, objectness,