From 02a9d1fd907cbcb66552e34827ecda05dca72741 Mon Sep 17 00:00:00 2001 From: Filip Drapejkowski Date: Mon, 6 Mar 2023 16:25:10 +0100 Subject: [PATCH] Bugfix float to int conversion - float point coordinates crashed cv2.line --- src/data_process/kitti_bev_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/data_process/kitti_bev_utils.py b/src/data_process/kitti_bev_utils.py index 1585400..1d83b50 100644 --- a/src/data_process/kitti_bev_utils.py +++ b/src/data_process/kitti_bev_utils.py @@ -164,7 +164,7 @@ def drawRotatedBox(img, x, y, w, l, yaw, color): bev_corners = get_corners(x, y, w, l, yaw) corners_int = bev_corners.reshape(-1, 1, 2).astype(int) cv2.polylines(img, [corners_int], True, color, 2) - corners_int = bev_corners.reshape(-1, 2) + corners_int = bev_corners.reshape(-1, 2).astype(int) cv2.line(img, (corners_int[0, 0], corners_int[0, 1]), (corners_int[3, 0], corners_int[3, 1]), (255, 255, 0), 2)