2503eba173
Fix memory leak
...
This commit moves cublasCreate out from dcn_v2_cuda_forward
to save some milliseconds and it adds cublasDestroy (cause
of memory leak).
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-06 23:09:12 +01:00
70eb5214cc
Change CenterNet input dimension.
...
This commit changes the image input dimension, it updates the
CenterNet detection class.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-06 18:08:06 +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
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
e99b353d8b
Fix the inference operation of the deformable convolutional layer.
...
This commit removes the malloc operation in the inference
method and adds the sigmoid kernel.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2019-12-19 14:47:57 +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
Davide Sapienza
xavier