Fix logger error in DeepStream 6.0 / 6.0.1 + Change output classes format + Fixes

2023-06-08 13:47:43 -03:00
parent 9fd80c5248
commit 64fa573f72
23 changed files with 233 additions and 258 deletions
--- a/utils/export_yoloV8.py
+++ b/utils/export_yoloV8.py
@@ -19,6 +19,7 @@ class DeepStreamOutput(nn.Module):
        x = x.transpose(1, 2)
        boxes = x[:, :, :4]
        scores, classes = torch.max(x[:, :, 4:], 2, keepdim=True)
+        classes = classes.float()
        return boxes, scores, classes