Update Benchmarks + Add YOLOv7-u6 + Fixes

2023-05-21 02:12:09 -03:00
parent af20c2f72c
commit 79d22283c1
13 changed files with 176 additions and 87 deletions
--- a/utils/export_yoloV5.py
+++ b/utils/export_yoloV5.py
@@ -19,7 +19,7 @@ class DeepStreamOutput(nn.Module):
        boxes = x[:, :, :4]
        objectness = x[:, :, 4:5]
        scores, classes = torch.max(x[:, :, 5:], 2, keepdim=True)
-        return torch.cat((boxes, scores, classes, objectness), dim=2)
+        return torch.cat((boxes, scores * objectness, classes), dim=2)


 def suppress_warnings():