fix: don't convert to int

artefactory · Nov 29, 2023 · 6d4ea07 · 6d4ea07
1 parent bc045ca
commit 6d4ea07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trackreid/tracked_object_metadata.py b/trackreid/tracked_object_metadata.py
@@ -49,7 +49,7 @@ def update(self, data_line: np.ndarray, frame_id: int):
 
         class_name = int(data_line[input_data_positions.category])
         self.class_counts[class_name] = self.class_counts.get(class_name, 0) + 1
-        self.bbox = list(map(int, data_line[input_data_positions.bbox]))
+        self.bbox = list(data_line[input_data_positions.bbox])
         confidence = float(data_line[input_data_positions.confidence])
         self.confidence = confidence
         self.confidence_sum += confidence
@@ -126,7 +126,7 @@ def to_dict(self):
             "first_frame_id": int(self.first_frame_id),
             "last_frame_id": int(self.last_frame_id),
             "class_counts": class_counts_str,
-            "bbox": [int(i) for i in self.bbox],
+            "bbox": [i for i in self.bbox],
             "confidence": float(self.confidence),
             "confidence_sum": float(self.confidence_sum),
             "observations": int(self.observations),