8968f3c8a2
* add box visualization * add box visualization and change thresholds for pif preprocessing * refactor printer * change default values * change confidence definition * remove redundant function * add debug plot in preprocessing * add task error in evaluation * add horizontal flipping * add evaluation table * add evaluation table with verbosity * add tabulate requirement and command line option verbose * refactor evaluate * add task error with mean absolute deviation * add stereo baseline * integrate stereo baseline * refactor factory preprocessing * add stereo command for evaluation * fix category bug * add interquartile range for stereo * use left tt for translation * refactor stereo functions * remvove redundant functions * change names of constants * add pixel error as function of depth * fix bug on output directory * add now time at the moment of saving * add person sitting category * remove box in pifpaf predictions * fix printing name * add printing of number of matches * add cyclist category * fix assertion error * add travis file * working eval * working eval * change source file * renaming * add pylint file * fix pylint * fix import * add pyc files in gitignore * pylint fix * pylint fix * add pytest cache * update readme * fix pylint * fix pylint * add travis file * add pylint in pip install * fix pylint
25 lines
774 B
Python
25 lines
774 B
Python
import os
|
|
import sys
|
|
|
|
# Python does not consider the current directory to be a package
|
|
sys.path.insert(0, os.path.join('..', 'monoloco'))
|
|
|
|
|
|
def test_iou():
|
|
from monoloco.utils.iou import get_iou_matrix
|
|
boxes_pred = [[1, 100, 1, 200]]
|
|
boxes_gt = [[100., 120., 150., 160.],[12, 110, 130., 160.]]
|
|
iou_matrix = get_iou_matrix(boxes_pred, boxes_gt)
|
|
assert iou_matrix.shape == (len(boxes_pred), len(boxes_gt))
|
|
|
|
|
|
def test_pixel_to_camera():
|
|
from monoloco.utils.camera import 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
|