6a68f19b2c
Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 ( Fix #74 )
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Francesco Gatti <gattifrancesco@hotmail.it >
2020-07-16 19:06:54 +02:00
c4aad7fe95
Patch for CUDNN 8.0.1
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-16 18:16:09 +02:00
4361d5fec0
Remove the final parameter from the layers
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 18:48:03 +02: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
4616be0738
Add grouped convolutions in CUDNN and tensorRT.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-05 14:32:58 +01:00
0f0c8c28e1
Merge with master, works with Jetpack 4.3
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-01-15 19:24:19 +01:00
c32a0be257
Batchnorm eps fix, works on jetpack 4.3
2020-01-15 18:06:02 +01:00
7e7f480e2b
Yolov3_tiny works on tensorRT
...
Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
d889ed385d
CenterNet TensorRT works. TensorRT serialization not yet implemented
...
Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com >
2019-12-20 11:05:03 +01:00
df888a3457
Add CenterNet based on Resnet101, TensorRT not implemented.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2019-12-18 18:59:54 +01:00
2594f59d0d
conv2d ok, but deconv ha different dim with tensorrt
2019-10-30 16:45:46 +01:00
f9afee2f3b
deconv layer cudnn
2019-10-30 00:05:22 +01:00
8c629ebe7b
string input and flir test
2019-09-14 19:03:13 +02:00
6249956469
namespace change
2018-12-14 21:55:16 +01:00
266330009c
opencv viz
2017-08-10 16:22:17 +02:00
4526e2767a
NetworkRT (deallocations to be done)
2017-08-03 12:16:57 +02:00
e8355cee67
better network model
2017-08-01 23:03:02 +02:00
ed5e5d58b5
TensorRT version
2017-08-01 17:51:49 +02:00
b94931f9f7
yolo layers
2017-08-01 16:08:56 +02:00
e4df86a07c
cudnn5 branch for tx2
2017-07-14 11:11:51 +02:00
cfa43a8580
conv2d destry
2017-06-28 15:29:14 +00:00
3cb126420c
conv2d implementation
2017-06-28 14:04:43 +00:00
Micaela Verucchi
Davide Sapienza
fbagni
Francesco Gatti
mbosi