diff --git a/utils/export_damoyolo.py b/utils/export_damoyolo.py
index ff0ebbf..01f69ea 100644
--- a/utils/export_damoyolo.py
+++ b/utils/export_damoyolo.py
@@ -18,7 +18,7 @@ class DeepStreamOutput(nn.Module):
     def forward(self, x):
         boxes = x[1]
         scores, classes = torch.max(x[0], 2, keepdim=True)
-        return torch.cat((boxes, scores, classes), dim=2)
+        return torch.cat((boxes, scores, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yoloV5.py b/utils/export_yoloV5.py
index ca2703c..f5ca1d1 100644
--- 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 * objectness, classes), dim=2)
+        return torch.cat((boxes, scores * objectness, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yoloV6.py b/utils/export_yoloV6.py
index 6b224f3..6daeb27 100644
--- a/utils/export_yoloV6.py
+++ b/utils/export_yoloV6.py
@@ -23,7 +23,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 * objectness, classes), dim=2)
+        return torch.cat((boxes, scores * objectness, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yoloV7.py b/utils/export_yoloV7.py
index d33b001..1dca872 100644
--- a/utils/export_yoloV7.py
+++ b/utils/export_yoloV7.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 * objectness, classes), dim=2)
+        return torch.cat((boxes, scores * objectness, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yoloV7_u6.py b/utils/export_yoloV7_u6.py
index d1fd5f5..d0a1864 100644
--- a/utils/export_yoloV7_u6.py
+++ b/utils/export_yoloV7_u6.py
@@ -18,7 +18,7 @@ class DeepStreamOutput(nn.Module):
         x = x.transpose(1, 2)
         boxes = x[:, :, :4]
         scores, classes = torch.max(x[:, :, 4:], 2, keepdim=True)
-        return torch.cat((boxes, scores, classes), dim=2)
+        return torch.cat((boxes, scores, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yoloV8.py b/utils/export_yoloV8.py
index 12d94f1..fe441ce 100644
--- a/utils/export_yoloV8.py
+++ b/utils/export_yoloV8.py
@@ -19,7 +19,7 @@ class DeepStreamOutput(nn.Module):
         x = x.transpose(1, 2)
         boxes = x[:, :, :4]
         scores, classes = torch.max(x[:, :, 4:], 2, keepdim=True)
-        return torch.cat((boxes, scores, classes), dim=2)
+        return torch.cat((boxes, scores, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yolonas.py b/utils/export_yolonas.py
index d5ca4b8..df7983d 100644
--- a/utils/export_yolonas.py
+++ b/utils/export_yolonas.py
@@ -15,7 +15,7 @@ class DeepStreamOutput(nn.Module):
     def forward(self, x):
         boxes = x[0]
         scores, classes = torch.max(x[1], 2, keepdim=True)
-        return torch.cat((boxes, scores, classes), dim=2)
+        return torch.cat((boxes, scores, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yolor.py b/utils/export_yolor.py
index 33d6960..86fefec 100644
--- a/utils/export_yolor.py
+++ b/utils/export_yolor.py
@@ -16,7 +16,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 * objectness, classes), dim=2)
+        return torch.cat((boxes, scores * objectness, classes.float()), dim=2)
 
 
 def suppress_warnings():
diff --git a/utils/export_yolox.py b/utils/export_yolox.py
index 3cdd8f3..75d2f0a 100644
--- a/utils/export_yolox.py
+++ b/utils/export_yolox.py
@@ -18,7 +18,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 * objectness, classes), dim=2)
+        return torch.cat((boxes, scores * objectness, classes.float()), dim=2)
 
 
 def suppress_warnings():