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_ppyoloe.py
+++ b/utils/export_ppyoloe.py
@@ -19,7 +19,7 @@ class DeepStreamOutput(nn.Layer):
        boxes = x['bbox']
        x['bbox_num'] = x['bbox_num'].transpose([0, 2, 1])
        scores = paddle.max(x['bbox_num'], 2, keepdim=True)
-        classes = paddle.argmax(x['bbox_num'], 2, keepdim=True)
+        classes = paddle.cast(paddle.argmax(x['bbox_num'], 2, keepdim=True), dtype='float32')
        return boxes, scores, classes