8c1eb69494
Merge with master works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 14:59:42 +02:00
34968eb4ad
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +02:00
b070bb3bd8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-08 11:20:56 +02:00
6a08118e3d
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +02:00
5be71cf9b7
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
3a5115578e
Fix the INT8 calibrator sintax
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 10:22:03 +02:00
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
537b0cfa21
Fix the sorting kernels used in the CenterNet pre and post-processing.
...
This commit moves the kernels in the correct sub-directory.
It creates new header file for Thrust kernels. It splits the
kernels into two files: 'normalize.cu' contains CenterNet
pre-processing operations, 'postprocessing.cu' contains the
CenterNet post-processing operations.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-07 18:28:12 +02:00
52e1a192e3
Fix the tensorRT code to support versions prior to 6.0
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 19:54:40 +02:00
c63ff6ecd2
Fix compile errors dut to Int8BatchStream
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 19:11:39 +02:00
769284574b
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-06 18:13:02 +02:00
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
2fbd24625c
Move the calibration table saving to the tests/network folder.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 18:06:21 +02:00
92dae7f5c5
Add opencv script, modified macro, add opencv section into readme
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:25:53 +02:00
bbef698ea3
Fix getNetworkRTName (works also on Xavier)
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-02 10:15:30 +02:00
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
03d2b88fa6
Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable
...
This commit adds two variables for the calibration dataset.
The first is reffered to .txt file that contains the list
of the absolute paths of the images for the INT8 calitration.
The second is referred to .txt file that contains the list
of the absolute paths of the labels of the same images above.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 18:55:32 +02:00
0474f019cf
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
e540213da6
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
785a3d8dd5
Add GPU version for yolo3detection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 21:29:29 +01:00
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
8b9516da97
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
997e857e32
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
5c6d6ae0af
Style fix, useless params removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 18:13:35 +01:00
cae0b84687
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
1a50065bde
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
14d850751d
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
677b6b2698
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
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
a65c8aa91e
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
4845edbc88
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +01:00
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
98770ee0c5
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
07a61b0e37
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
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
6fee8d5ec5
- ImuOdom model into class
...
- Route layer input array hard copy
- fix utils
2020-03-09 19:44:21 +01:00
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
1d38d88c85
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +01:00
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
c4cef1aa02
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-21 10:49:35 +01:00
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
cd1e1c1582
LSTM ok
2020-02-16 17:21:58 +01:00
4b48a2f38c
structure ok, result wrong
2020-02-16 17:08:19 +01:00
0ac292ea48
works but it need cleaning
2020-02-16 16:28:39 +01:00
8736a8c3da
LSTM params
2020-02-15 20:37:08 +01:00
200e9466f6
Move pre-processing on GPU
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-14 18:44:13 +01:00
f5706d8523
lstm return seq
2020-02-13 23:21:28 +01:00
0329318ca3
removed unused var
2020-02-13 23:10:48 +01:00
e8280d3a65
LSTM to be tested
2020-02-13 23:04:29 +01:00
c876fa05ae
LSTM cudnn test
2020-02-13 19:27:18 +01:00
Micaela Verucchi
Davide Sapienza
Micaela Verucchi
Francesco Gatti