Merge pull request #21 from peabody124/numpy_int_issue

np.float -> float
peabody124 · Apr 6, 2023 · c987d43 · c987d43
2 parents 16f9b02 + b2b8dfd
commit c987d43
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pose_pipeline/wrappers/deep_sort_yolov4/deep_sort/detection_yolo.py b/pose_pipeline/wrappers/deep_sort_yolov4/deep_sort/detection_yolo.py
@@ -27,7 +27,7 @@ class Detection_YOLO(object):
     """
 
     def __init__(self, tlwh, confidence, cls):
-        self.tlwh = np.asarray(tlwh, dtype=np.float)
+        self.tlwh = np.asarray(tlwh, dtype=float)
         self.confidence = float(confidence)
         self.cls = cls
 

diff --git a/pose_pipeline/wrappers/deep_sort_yolov4/deep_sort/preprocessing.py b/pose_pipeline/wrappers/deep_sort_yolov4/deep_sort/preprocessing.py
@@ -37,7 +37,7 @@ def non_max_suppression(boxes, max_bbox_overlap, scores=None):
     if len(boxes) == 0:
         return []
 
-    boxes = boxes.astype(np.float)
+    boxes = boxes.astype(float)
     pick = []
 
     x1 = boxes[:, 0]