Skip to content

Commit

Permalink
fix a bug for yolox
Browse files Browse the repository at this point in the history
  • Loading branch information
HaoyiZhu committed Jun 25, 2022
1 parent d7f73bd commit 4e55942
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion detector/yolox/yolox/utils/boxes.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def postprocess(
detections = detections[nms_out_index]
batch_idx = detections.new(detections.size(0), 1).fill_(i)
detections = torch.cat((batch_idx, detections), 1)
if not output:
if isinstance(output, int) and output == 0:
output = detections
else:
output = torch.cat((output, detections))
Expand Down

0 comments on commit 4e55942

Please sign in to comment.