Commit Graph

18 Commits

Author SHA1 Message Date
Davide Sapienza 0d6acaec41 Add grouped convolutions in CUDNN and tensorRT.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-05 14:32:58 +01:00
xavier 0f0c8c28e1 Merge with master, works with Jetpack 4.3
Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-01-15 19:24:19 +01:00
xavier c32a0be257 Batchnorm eps fix, works on jetpack 4.3 2020-01-15 18:06:02 +01:00
fbagni 7e7f480e2b Yolov3_tiny works on tensorRT
Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
Davide Sapienza 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
Davide Sapienza 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
Francesco Gatti 2594f59d0d conv2d ok, but deconv ha different dim with tensorrt 2019-10-30 16:45:46 +01:00
Francesco Gatti f9afee2f3b deconv layer cudnn 2019-10-30 00:05:22 +01:00
mbosi 8c629ebe7b string input and flir test 2019-09-14 19:03:13 +02:00
Francesco Gatti 6249956469 namespace change 2018-12-14 21:55:16 +01:00
Francesco Gatti 266330009c opencv viz 2017-08-10 16:22:17 +02:00
Francesco Gatti 4526e2767a NetworkRT (deallocations to be done) 2017-08-03 12:16:57 +02:00
Francesco Gatti e8355cee67 better network model 2017-08-01 23:03:02 +02:00
Francesco Gatti ed5e5d58b5 TensorRT version 2017-08-01 17:51:49 +02:00
Francesco Gatti b94931f9f7 yolo layers 2017-08-01 16:08:56 +02:00
Francesco Gatti e4df86a07c cudnn5 branch for tx2 2017-07-14 11:11:51 +02:00
Francesco Gatti cfa43a8580 conv2d destry 2017-06-28 15:29:14 +00:00
Francesco Gatti 3cb126420c conv2d implementation 2017-06-28 14:04:43 +00:00