Davide Sapienza
97b88ef52d
Fix a bug in the deformable kernel.
...
There was a wrong variable initialization.
Fixes: 51ffcb1f50 ("Optimize deformable kernel")
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-11 16:28:17 +01:00
Davide Sapienza
51ffcb1f50
Optimize deformable kernel
...
Signed-off.by: Ignacio Sañudo Olmedo<ignacio.sanudoolmedo@unimore.it >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-11 10:06:08 +01:00
Davide Sapienza
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
Davide Sapienza
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
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
Micaela Verucchi
8b64d1876f
deformable conv cu
2019-10-30 10:20:04 +01:00