handle ditance None type

This commit is contained in:
2024-07-04 18:45:00 +02:00
parent 76db5bd3d5
commit e2ff72a457
+3 -2
View File
@@ -295,8 +295,9 @@ if __name__ == '__main__':
i = 0 i = 0
for ct in tracker_list: for ct in tracker_list:
mean_distance = ct.process(image, image_anno) mean_distance = ct.process(image, image_anno)
cv2.putText(image_anno, f"Distance [{i}] : ({mean_distance[0]:+05.2f}, {mean_distance[1]:+05.2f})", if mean_distance is not None:
(20, 25 + 25 * i), cv2.FONT_HERSHEY_SIMPLEX, 0.5, ct.color, 1) cv2.putText(image_anno, f"Distance [{i}] : ({mean_distance[0]:+05.2f}, {mean_distance[1]:+05.2f})",
(20, 25 + 25 * i), cv2.FONT_HERSHEY_SIMPLEX, 0.5, ct.color, 1)
i += 1 i += 1
cv2.imshow(f"Image Anno", image_anno) cv2.imshow(f"Image Anno", image_anno)
else: else: