Commit Graph

393 Commits

Author SHA1 Message Date
Francesco Gatti 555b32b5dc Refactoring
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-14 17:22:58 +01:00
Francesco Gatti cf7fbadcd2 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 5e0c1879bc Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-13 12:17:21 +01:00
xavier 32dd807ab0 Add Yolov3_tiny512 test cpp
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-13 12:16:18 +01:00
Micaela Verucchi 8aa792925d Update README.md 2020-03-12 20:42:44 +01:00
xavier 9d17cb42f4 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 20:18:11 +01:00
xavier db10ebeb34 Fix print on times.csv
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-12 20:18:02 +01:00
Francesco Gatti 62e7882744 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 18:43:33 +01:00
Francesco Gatti 13b4dffd71 csresnext50-panet-spp works with TensorRT
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-12 18:24:41 +01:00
Francesco Gatti f6527f51e3 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 66e3b98b04 Update README.md 2020-03-11 12:04:28 +01:00
Micaela Verucchi 0cc2666e26 Update README.md 2020-03-11 11:51:28 +01:00
xavier d0d462c015 Merge
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-10 15:43:26 +01:00
xavier 2d46f16b70 Add tiny yolo 512 test
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-10 15:40:11 +01:00
Micaela Verucchi e6aa73d7bc Update README.md 2020-03-10 15:26:57 +01:00
Micaela Verucchi 4b85a2238a 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 8944778dcf imuodom fix 2020-03-09 22:05:34 +01:00
Francesco Gatti 406f8cc9b3 - ImuOdom model into class
- Route layer input array hard copy
- fix utils
2020-03-09 19:44:21 +01:00
Micaela Verucchi 7ea31d123a Update README
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-03-08 18:12:08 +01:00
Davide Sapienza 7dbbbe4d9a Update README
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-03-03 17:34:01 +01:00
Davide Sapienza 01a42ffe27 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 296a6cbc87 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 fe206ea24c Fix dependencies problems
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-02-27 16:48:10 +01:00
xavier 3eb079dd13 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 110bf56dc4 Fix include error
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-21 11:14:52 +01:00
xavier 209a3e8492 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-21 10:49:35 +01:00
xavier 38a1b9dcb2 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 443691414a LSTM ok 2020-02-16 17:21:58 +01:00
Francesco Gatti 1a1c54f364 structure ok, result wrong 2020-02-16 17:08:19 +01:00
Francesco Gatti 10b7160677 works but it need cleaning 2020-02-16 16:28:39 +01:00
Francesco Gatti 4746121d43 LSTM params 2020-02-15 20:37:08 +01:00
Davide Sapienza 2c1df5619f Move pre-processing on GPU
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-14 18:44:13 +01:00
Francesco Gatti 4fa5d2c231 lstm return seq 2020-02-13 23:21:28 +01:00
Francesco Gatti c1c2173e4d removed unused var 2020-02-13 23:10:48 +01:00
Francesco Gatti 03d39d991c LSTM to be tested 2020-02-13 23:04:29 +01:00
Francesco Gatti a9c0db0bf6 LSTM cudnn test 2020-02-13 19:27:18 +01:00
Davide Sapienza 97b88ef52d 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 4c2d2a7965 Update README.md 2020-02-11 15:12:27 +01:00
xavier 636d899f9f Merge README
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 14:58:25 +01:00
xavier 7e21b10aee Add script to download COCO val2017
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 14:56:54 +01:00
Micaela Verucchi 6e2ff405f4 Update README.md 2020-02-11 11:28:20 +01:00
xavier 1936e54870 Change README.md, add config.yaml, refactoring
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 11:25:42 +01:00
xavier b5c0baa99d Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-11 10:32:52 +01:00
xavier 801b8b5641 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 51ffcb1f50 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 10831ab450 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-10 18:19:12 +01:00
xavier d601e980f6 Add avg precision, recall, f1score computation, other minors
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-10 18:19:08 +01:00
Davide Sapienza b06dc286e9 Update README.md
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-10 11:19:39 +01:00
Davide Sapienza c02238ddc8 Add Anaconda environment for ResNet101 and DLA34 weights exporter
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-10 10:58:04 +01:00
Davide Sapienza e72aa348a0 Add DLA34 weights exporter
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-10 10:56:33 +01:00