b75cecb105
Added constant and zero padding with ISliceLayer,need to add them for tensorrt versions less than 8.2 using IPluginV2Ext instead of ISliceLayer since they dont seem to support reflect and zero
2022-01-04 09:03:04 +05:30
ba022663f1
completed padding migrations from github
2022-01-03 19:49:32 +05:30
6837644eb2
add reflection padding from github
2022-01-03 18:58:13 +05:30
d2d44e9e92
Fix max elem (remove thrust) for segmentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-05 19:54:02 +05:30
0292e21c33
Merge remote-tracking branch 'origin/master' into cnet
2021-07-22 17:03:23 +02:00
84ff978ccb
Merge with master, all tests passed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 12:48:41 +02:00
28fab9c3e1
Merge remote-tracking branch 'origin/master' into cnet
2021-04-27 14:43:39 +02:00
6aa8666be5
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
7f65ee0b2a
Merge remote-tracking branch 'origin/master' into cnet
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-07 17:45:44 +01:00
d25803d438
Post-processing improved
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-24 18:46:25 +02:00
94e558003d
Shelfnet works on tensorRT (shortcut need to be fixed)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-23 12:50:24 +02:00
6fd261f628
Shelfnet works on cuDNN. To test everything else
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-22 18:11:19 +02:00
9f1e30eaa9
Add shelfnet. Resnet18backbone works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-19 18:55:42 +02:00
d936e5f740
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
ba8c282384
Add 3D CenterNet detection class
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:03:09 +02:00
db0f8a4d99
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
6e5ab031c4
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
0e5c90634f
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
cf3fbeddbd
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
f6527f51e3
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
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
97b88ef52d
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
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
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
7f239efdc0
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
perseusdg
Micaela Verucchi
Davide Sapienza
hchandirasekar
dsapienza
Francesco Gatti
fbagni