0180f7830b
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
40aefba507
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
26d8d009c5
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
5be71cf9b7
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
537b0cfa21
Fix the sorting kernels used in the CenterNet pre and post-processing.
...
This commit moves the kernels in the correct sub-directory.
It creates new header file for Thrust kernels. It splits the
kernels into two files: 'normalize.cu' contains CenterNet
pre-processing operations, 'postprocessing.cu' contains the
CenterNet post-processing operations.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-07 18:28:12 +02:00
5f4508a5a6
Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-11 18:37:26 +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
40a4a55cd0
Fix a bug in the deformable kernel.
...
There was a wrong variable initialization.
Fixes: 51ffcb1f50sapienza.dav@gmail.com >
2020-02-11 16:28:17 +01:00
2b06836a7f
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
5abf1613f2
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
07da54efe9
Centernet: fix pooling problem, add centrnet demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-21 12:50:18 +01:00
7e7f480e2b
Yolov3_tiny works on tensorRT
...
Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
f0ea2c027f
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2019-12-20 15:27:55 +01:00
854a4c316a
Add ResizeLayerRT plugin
...
Signed-off-by: nvidia <micaelaverucchi@gmail.com >
2019-12-20 15:27:41 +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
8b64d1876f
deformable conv cu
2019-10-30 10:20:04 +01:00
c8f2e1b448
upsample ok
2018-12-20 17:08:31 +01:00
e91db28756
shortcut cu
2018-12-20 09:53:17 +01:00
2cf8d8f6fc
fp16 implementation, TODO deallocate in LayerWgs
2017-08-30 14:37:25 +00:00
3124f86878
stream in TRT plugin
2017-08-10 19:21:15 +02:00
9a6058ac4a
removed sync
2017-08-10 18:47:21 +02:00
266330009c
opencv viz
2017-08-10 16:22:17 +02:00
ed5e5d58b5
TensorRT version
2017-08-01 17:51:49 +02:00
b94931f9f7
yolo layers
2017-08-01 16:08:56 +02:00
0a0acb9b1d
test ready
2017-06-29 12:26:51 +00:00
b9f894ca75
better elu kernel
2017-06-28 15:16:38 +00:00
3cb126420c
conv2d implementation
2017-06-28 14:04:43 +00:00
a86df107be
ELU implemented
2017-06-28 11:56:43 +00:00
hchandirasekar
Micaela Verucchi
dsapienza
Davide Sapienza
Francesco Gatti
fbagni