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
9007e25a00
Remove mallocs and frees from the kernels
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-07 14:50:05 +01:00
289a97d06c
Refactoring map_demo, add evaluation.h and evaluation.cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 12:47:02 +01:00
5c501f529b
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-07 09:06:29 +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
c21a0687ce
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-06 18:24:49 +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
37019597af
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-05 18:39:32 +01:00
d8eeb36d4b
Add mAP computation and demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-05 18:39:13 +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
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
c68d6f318e
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-01-20 16:41:19 +01:00
b23cf9fe70
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
...
Signed-off-by: Michaela Verucchi <micaelaverucchi@gmail.com >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-20 12:35:27 +01:00
7838cb4922
Pre-process, Process and Post-process work
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-20 12:27:49 +01:00
da4f246157
add DLA, plugin for shortcut and leaky. new verison 0.4
2020-01-15 21:48:18 +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
5272f1cde6
CenterNet TensorRT serialization works
...
This commit adds the Deformable layer serialization.
Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com >
2019-12-23 15:31:13 +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
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
44b2bce3ff
Yolo3_tiny CUDNN works, TensorRT doesn't. Add n_masks to Yolo layer.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2019-12-05 20:43:48 +00:00
2594f59d0d
conv2d ok, but deconv ha different dim with tensorrt
2019-10-30 16:45:46 +01:00
6a1707f65e
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2019-10-30 10:20:44 +01:00
8b64d1876f
deformable conv cu
2019-10-30 10:20:04 +01:00
33f36ab204
Deconv tensorrt
2019-10-30 09:40:29 +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
42a1ea02b9
resnet merge
2019-10-28 18:38:23 +01:00
ca62784f57
include dir fix, cmake dir
2019-09-17 15:22:39 +02:00
ec02c7292f
save layer names in rt file
2019-09-16 19:41:59 +02:00
8c629ebe7b
string input and flir test
2019-09-14 19:03:13 +02:00
3714155809
dla
2019-03-06 16:40:12 +01:00
7505c28d2d
include fix
2019-02-19 11:09:27 +00:00
c7941666ec
demo for more yolo3
2019-02-19 08:43:35 +00:00
87fe342ca2
yoloRT load anchors
2019-02-18 21:39:14 +01:00
bdd8e0bc26
yolo3plug fix
2019-02-18 18:55:48 +00:00
738fa94150
version update
2019-02-18 15:54:22 +00:00
13063b904d
yolo3 ok
2019-02-18 15:51:57 +00:00
0d682136de
yolo3 berkeley ok
2019-02-18 15:37:39 +00:00
2c63bf05be
multipl yolo morge
2019-02-06 22:24:01 +00:00
0e97452460
dects dont works
2019-02-05 20:09:47 +00:00
c8dea4668d
compute detections
2019-02-04 20:34:15 +00:00
2e8d0b1002
yolo3 86 route error
2018-12-23 16:20:17 +01:00
3bd725801d
upsample ok, route have problems
2018-12-22 23:56:01 +01:00
34be4cd00f
yoloRT layer
2018-12-22 21:26:50 +01:00
64626bf547
shortcut rt test
2018-12-21 15:53:39 +01:00
c8f2e1b448
upsample ok
2018-12-20 17:08:31 +01:00
Davide Sapienza
xavier
Francesco Gatti
fbagni
mbosi
Roberto Cavicchioli