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
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
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
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
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
f9afee2f3b
deconv layer cudnn
2019-10-30 00:05:22 +01:00
02936fa928
ResNet working
2019-10-29 17:21:31 +01:00
e9ec582223
Shortcat ok
2019-10-29 15:08:44 +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
2cf8d8f6fc
fp16 implementation, TODO deallocate in LayerWgs
2017-08-30 14:37:25 +00:00
e8355cee67
better network model
2017-08-01 23:03:02 +02:00
1cfe70365f
yolo test
2017-08-01 17:12:29 +02:00
b94931f9f7
yolo layers
2017-08-01 16:08:56 +02:00
0767df43a2
initial commit
2017-06-28 01:14:39 +02:00
Micaela Verucchi
Davide Sapienza
Francesco Gatti
mbosi