diff --git a/tests/test_utils.py b/tests/test_utils.py index 9604076..6a4f58d 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -1,6 +1,7 @@ from utils.iou import get_iou_matrix +from utils.camera import pixel_to_camera def test_iou(): @@ -9,3 +10,12 @@ def test_iou(): iou_matrix = get_iou_matrix(boxes_pred, boxes_gt) assert iou_matrix.shape == (len(boxes_pred), len(boxes_gt)) + +def test_pixel_to_camera(): + kk = [[718.3351, 0., 600.3891], [0., 718.3351, 181.5122], [0., 0., 1.]] + zz = 10 + uv_vector = [1000., 400.] + xx_norm = pixel_to_camera(uv_vector, kk, 1)[0] + xx_1 = xx_norm * zz + xx_2 = pixel_to_camera(uv_vector, kk, zz)[0] + assert xx_1 == xx_2