change np.float to np.float32

levan92 · Jan 16, 2023 · 043cb5e · 043cb5e
1 parent 2491e6d
commit 043cb5e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deep_sort_realtime/deep_sort/detection.py b/deep_sort_realtime/deep_sort/detection.py
@@ -34,7 +34,7 @@ class Detection(object):
 
     def __init__(self, ltwh, confidence, feature, class_name=None, instance_mask=None, others=None):
         # def __init__(self, ltwh, feature):
-        self.ltwh = np.asarray(ltwh, dtype=np.float)
+        self.ltwh = np.asarray(ltwh, dtype=np.float32)
         self.confidence = float(confidence)
         self.feature = np.asarray(feature, dtype=np.float32)
         self.class_name = class_name

diff --git a/deep_sort_realtime/utils/nms.py b/deep_sort_realtime/utils/nms.py
@@ -28,7 +28,7 @@ def non_max_suppression(boxes, max_bbox_overlap, scores=None):
     if len(boxes) == 0:
         return []
 
-    boxes = boxes.astype(np.float)
+    boxes = boxes.astype(np.float32)
     pick = []
 
     x1 = boxes[:, 0]