Commit Graph

20 Commits

Author SHA1 Message Date
Micaela Verucchi 6a08118e3d Refactoring & documentation
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-08 11:20:48 +02:00
Micaela Verucchi 6e55e2376f Move extraction of name into function in utils
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-08 10:09:15 +02:00
Micaela Verucchi 065c4e2b58 Minor fix on output file name in map demo
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-08 09:35:54 +02:00
Micaela Verucchi 733755e7e0 Add evaluation to tk::dnn namespace, style fix also
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-06 18:12:38 +02:00
Micaela Verucchi 43567dc3ea Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-01 19:18:43 +02:00
Micaela Verucchi f4b976c793 Add tp tests for yolo3512 and yolo3tiny512
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-26 15:06:49 +01:00
Micaela Verucchi 3bc648c07b Modify map demo, using abstract class. Move draw function in abstract class
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-23 19:32:37 +01:00
Micaela Verucchi 8b9516da97 Refactoring for detection NN
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-20 21:14:12 +01:00
Micaela Verucchi f6129311d6 Add support for mobilenet in map_demo
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-17 15:08:43 +01:00
Francesco Gatti 21b8146232 Add function to download weigths if do no exist, for each test. Add some controls in map demo.
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-14 17:02:07 +01:00
xavier 5c347b1e11 Fix print on times.csv
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-12 20:18:02 +01:00
Micaela Verucchi 75d5e20843 Add writing results on file for map demo
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-10 14:35:25 +01:00
xavier 808a84131c Add Mobilenet2SSDLite test
The new test works both with TensorRT and cuDNN. Preprocessing and
Postprocessing are missing. Add ClippedReLU (for ReLU6), groups for
Conv2d, additional bias for convolution.

Other minors:
-move the timer in the detector to measure all the
processing time for a given frame (both centernet and yolo);
-add int8 flag.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
Davide Sapienza <sapienza.dav@gmail.com>
2020-02-21 10:45:46 +01:00
xavier 9c4ea524fc Add script to download COCO val2017
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 14:56:54 +01:00
xavier d0026ff235 Change README.md, add config.yaml, refactoring
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 11:25:42 +01:00
xavier 73548b179e Read parameters for mAP from yaml, add yampl-cpp dependency
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 10:32:42 +01:00
xavier c7cb3e7571 Add avg precision, recall, f1score computation, other minors
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-10 18:19:08 +01:00
xavier da24c2e752 Refactoring map_demo, add evaluation.h and evaluation.cpp
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-07 12:47:02 +01:00
xavier 9bc4ba0eff Add mAP 0.5:0.95, other small fix
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-07 09:09:40 +01:00
xavier ef18be7b7b Add mAP computation and demo
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-05 18:39:13 +01:00