Commit Graph

346 Commits

Author SHA1 Message Date
Micaela Verucchi 4a2a244162 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00
Francesco Gatti a7c3761960 eigen assert fix 2020-03-14 19:50:55 +01:00
Micaela Verucchi 94971fbdcb Fix download weights yolov3 versions
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-03-14 18:20:32 +01:00
Francesco Gatti 677b6b2698 Refactoring
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-14 17:22:58 +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 01995abdf8 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-13 12:17:21 +01:00
xavier 51b9780baf Add Yolov3_tiny512 test cpp
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-13 12:16:18 +01:00
Micaela Verucchi 35bbda0ab1 Update README.md 2020-03-12 20:42:44 +01:00
xavier a65c8aa91e Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 20:18:11 +01:00
xavier 5c347b1e11 Fix print on times.csv
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-12 20:18:02 +01:00
Francesco Gatti ef1ebea5e9 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 18:43:33 +01:00
Francesco Gatti 4845edbc88 csresnext50-panet-spp works with TensorRT
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-12 18:24:41 +01:00
Francesco Gatti 5f4508a5a6 Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-11 18:37:26 +01:00
Micaela Verucchi 963b2c40d1 Update README.md 2020-03-11 12:04:28 +01:00
Micaela Verucchi d5722cae3f Update README.md 2020-03-11 11:51:28 +01:00
xavier 98770ee0c5 Merge
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-10 15:43:26 +01:00
xavier 07a61b0e37 Add tiny yolo 512 test
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-10 15:40:11 +01:00
Micaela Verucchi 0c5e31476e Update README.md 2020-03-10 15:26:57 +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
Francesco Gatti 2d378e9070 imuodom fix 2020-03-09 22:05:34 +01:00
Francesco Gatti 6fee8d5ec5 - ImuOdom model into class
- Route layer input array hard copy
- fix utils
2020-03-09 19:44:21 +01:00
Micaela Verucchi 0773f216b3 Update README
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-03-08 18:12:08 +01:00
Davide Sapienza b734baca0c Update README
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-03-03 17:34:01 +01:00
Davide Sapienza 99e809737f Move multiple demo files into one
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-03-03 17:30:35 +01:00
Davide Sapienza c94159a2bf Add draw method to Yolo3Detection class.
Update yolo demo

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-03-03 17:08:25 +01:00
Micaela Verucchi 1d38d88c85 Fix dependencies problems
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-02-27 16:48:10 +01:00
xavier 3be254b08f Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-26 17:42:20 +01:00
xavier c36c898948 Fix include error
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-21 11:14:52 +01:00
xavier c4cef1aa02 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-21 10:49:35 +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
Francesco Gatti cd1e1c1582 LSTM ok 2020-02-16 17:21:58 +01:00
Francesco Gatti 4b48a2f38c structure ok, result wrong 2020-02-16 17:08:19 +01:00
Francesco Gatti 0ac292ea48 works but it need cleaning 2020-02-16 16:28:39 +01:00
Francesco Gatti 8736a8c3da LSTM params 2020-02-15 20:37:08 +01:00
Davide Sapienza 200e9466f6 Move pre-processing on GPU
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-14 18:44:13 +01:00
Francesco Gatti f5706d8523 lstm return seq 2020-02-13 23:21:28 +01:00
Francesco Gatti 0329318ca3 removed unused var 2020-02-13 23:10:48 +01:00
Francesco Gatti e8280d3a65 LSTM to be tested 2020-02-13 23:04:29 +01:00
Francesco Gatti c876fa05ae LSTM cudnn test 2020-02-13 19:27:18 +01:00
Davide Sapienza 40a4a55cd0 Fix a bug in the deformable kernel.
There was a wrong variable initialization.

Fixes: 51ffcb1f50 ("Optimize deformable kernel")

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-11 16:28:17 +01:00
Micaela Verucchi 217b3fbfb5 Update README.md 2020-02-11 15:12:27 +01:00
xavier b1a6b45f5f Merge README
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 14:58:25 +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
Micaela Verucchi 21535dd6b3 Update README.md 2020-02-11 11:28:20 +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 e01d7c6b13 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-11 10:32:52 +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
Davide Sapienza 2b06836a7f Optimize deformable kernel
Signed-off.by: Ignacio Sañudo Olmedo<ignacio.sanudoolmedo@unimore.it>
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-11 10:06:08 +01:00
xavier 3ef2cc65f5 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-10 18:19:12 +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