Commit Graph

17 Commits

Author SHA1 Message Date
Lorenzo Bertoni
019b6b0fad
make utils in torch and remove redundant functions (#3)
* Add precision metrics

* add mode gt_all and change default threshold

* add cyclists

* add iou matrix'
'

* add cyclists only in training phase

* add dropout in model name

* small typos

* small typo

* fix error on uv_boxes

* change default mode from gt_ped to gt

* 2 decimals

* fix name bug

* refactor prepare_pif_kps

* corrected get_keypoints_batch

* add pixel to camera for 3d vectors

* preprocessing in torch

* return original outputs

* Skeleton for post_process

* baseline version for post processing

* add keypoints torch in post_processing

* cleaning misc

* add reorder_matches

* update preprocess with get_iou_matches

* fix indices

* remove aa

* temp

* skeleton kitti_generate

* skeleton kitti_generate (2)

* refactor file

* remove old get_input_data

* refactor geometric eval

* refactor geometric eval(2)

* temp

* refactor geometric

* change saving order for txts

* update pixel to camera

* update depth

* Fix pixel to camera

* add xyz_from_distance

* use new function

* fix std_ale calculation in eval

* remove debug points
2019-06-28 18:33:58 +02:00
Lorenzo Bertoni
2aea30cb7d
add matrix IoU evaluation (#2)
* add boxes_3d and fix bug on kk

* delete stereo script

* add get_iou_matrix

* add statistics

* Add pytest

* add iou_matrix

* refactor evaluation kitti

* add box size constraint on moderate

* refactor evaluation

* pylint check
2019-06-24 08:45:39 +02:00
Lorenzo Bertoni
86438189a7
refactor predict (#1)
* add stereo arg (#1)

* stereo skeleton

* skeleton stereo (#2)

* skeleton stereo (#3)

* refactor for stereo

* stereo running

* modify stereo parameters

* remove warnings as errors

* add p2_right

* add factory file

* add p2_right

* add scaling factor to intrinsic matrix

* Refactor predict

* refactor predict 2

* temp

* add uppercase constants

* add uppercase constants

* working predict

* add todo

* Add person_sitting as option to uncomment

* turn off stereo

* remove stereo

* pylint corrections

* add break point

* temp

* fix small bug

* Add compatibility for pifpaf and monoloco networks

* pylint fix
2019-06-20 15:43:38 +02:00
lorenzo
6f3379e394 add factory method 2019-05-21 14:45:23 +02:00
lorenzo
6fbc496702 refactor __init (1) 2019-05-21 10:42:33 +02:00
lorenzo
738095bc8c add function for split_training 2019-05-21 10:30:21 +02:00
lorenzo
3c6c305606 remove class imports 2019-05-21 10:26:59 +02:00
lorenzo
d756678518 fix pylint(2) 2019-05-16 12:26:42 +02:00
lorenzo
3243da72cd fix pylint 2019-05-16 10:30:52 +02:00
lorenzo
2222d12e20 remove no-calib argument 2019-05-13 17:59:13 +02:00
lorenzo
cd551a17a8 Update readme 2019-05-13 15:08:29 +02:00
lorenzo
f72be64b8d add calb in dictioary definition 2019-05-13 14:13:51 +02:00
lorenzo
9eb4b3a5f5 add calibration in names.json 2019-05-13 13:59:31 +02:00
lorenzo
fb643e3350 add splits 2019-05-13 13:57:10 +02:00
lorenzo
3235e9d138 fix names of splits 2019-05-13 13:55:57 +02:00
lorenzo
cb344c4664 Update readme and fix data names 2019-05-13 12:46:58 +02:00
lorenzo
28faf348c5 first commit 2019-05-13 11:41:57 +02:00