charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add test for pixel_to_camera

Browse Source
This commit is contained in:
lorenzo 2019-07-02 11:28:56 +02:00
parent bf413019e0
commit 712bdf9dc6
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_utils.py
View File

@ -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