62e4a3f779
memory release
2020-06-02 12:43:06 +02:00
0458f361b1
release layer wgs and version update
2020-06-01 21:03:37 +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
0951f4644f
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
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
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
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
Francesco Gatti
Davide Sapienza
Micaela Verucchi
mbosi