ce20868bad
Add TKDNN_MODE variable to the name of the network.
...
This commit permits to obtain different .rt files for different
precision optimizations of the same network.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 15:08:25 +02:00
573924cf2c
Update README.md
2020-03-29 20:15:15 +02:00
f2a4125d9f
Update README.md
2020-03-29 20:07:27 +02:00
d7276c720d
Update README.md
2020-03-29 20:03:59 +02:00
482e122655
Fix opencv cuda include error
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 10:43:49 +01:00
08e1801c60
Add the Int8 calibrator and the tensorRT Int8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 00:48:14 +01:00
7c55dcb708
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
3eda9b9219
Add GPU version for yolo3detection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 21:29:29 +01:00
df37e11709
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
bbcc33c0cf
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
c7d9c38ea0
Add OPENCV_CUDA define, to allow having preprocess both in CPU and GPU (Mobilenet and Centernet)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 10:21:11 +01:00
58fe723b6a
Style fix, useless params removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 18:13:35 +01:00
f44f377771
Add preprocess function, allow preprocess on GPU for mobilenetdetection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 17:53:29 +01:00
41b1135fb3
Add support for mobilenet in map_demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 15:08:43 +01:00
43fd92f701
Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 13:18:05 +01:00
1d388ca51a
Add mobilenetv2ssdlite512 test.
...
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 10:35:02 +01:00
9c25d15ff2
Add yolov3_512 (size 512) test
2020-03-16 11:29:15 +01:00
62c8f528a4
eigen assert fix
2020-03-14 19:50:55 +01:00
82d907971c
Fix download weights yolov3 versions
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-14 18:20:32 +01:00
555b32b5dc
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
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
5e0c1879bc
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-13 12:17:21 +01:00
32dd807ab0
Add Yolov3_tiny512 test cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-13 12:16:18 +01:00
8aa792925d
Update README.md
2020-03-12 20:42:44 +01:00
9d17cb42f4
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
db10ebeb34
Fix print on times.csv
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-12 20:18:02 +01:00
62e7882744
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 18:43:33 +01:00
13b4dffd71
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +01:00
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
66e3b98b04
Update README.md
2020-03-11 12:04:28 +01:00
0cc2666e26
Update README.md
2020-03-11 11:51:28 +01:00
d0d462c015
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
2d46f16b70
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
e6aa73d7bc
Update README.md
2020-03-10 15:26:57 +01:00
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
8944778dcf
imuodom fix
2020-03-09 22:05:34 +01:00
406f8cc9b3
- ImuOdom model into class
...
- Route layer input array hard copy
- fix utils
2020-03-09 19:44:21 +01:00
7ea31d123a
Update README
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-08 18:12:08 +01:00
7dbbbe4d9a
Update README
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:34:01 +01:00
01a42ffe27
Move multiple demo files into one
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:30:35 +01:00
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
fe206ea24c
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +01:00
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
110bf56dc4
Fix include error
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-21 11:14:52 +01:00
209a3e8492
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-21 10:49:35 +01:00
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
443691414a
LSTM ok
2020-02-16 17:21:58 +01:00
1a1c54f364
structure ok, result wrong
2020-02-16 17:08:19 +01:00
10b7160677
works but it need cleaning
2020-02-16 16:28:39 +01:00
4746121d43
LSTM params
2020-02-15 20:37:08 +01:00
Davide Sapienza
Micaela Verucchi
Francesco Gatti
Micaela Verucchi