f32d8a859b
Add mAP 0.5:0.95, other small fix
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 09:09:40 +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
c695d8c5d7
Implement CenterNet based on DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-05 14:59:56 +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
fe85c26888
Implement DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-24 19:30:32 +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
8a4d1cac17
compile with tensorrt 5
2020-01-20 14:51:38 +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
146e144249
Update README.md
2020-01-16 18:24:35 +01:00
2f57ba1222
Update README.md
2020-01-16 18:21:34 +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
f3f5daf3db
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-01-15 18:07:44 +01:00
c2d73623e5
support clion
2020-01-15 18:07:40 +01:00
c32a0be257
Batchnorm eps fix, works on jetpack 4.3
2020-01-15 18:06:02 +01:00
57d7743f7e
Change opencv funcion call (due to OpenCV 4)
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-01-15 09:55:10 +01:00
23a1365dc4
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2019-12-23 16:14:24 +01:00
7e7f480e2b
Yolov3_tiny works on tensorRT
...
Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +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
6bf9179acc
fix to drivework global path
2019-12-12 12:30:24 +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
a2db98670a
Add Yolov3 (COCO80) and Yolov3-tiny (COCO80), TensorRT for tiny not working
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2019-12-04 17:01:40 +00:00
b218b18a02
readme update
2019-12-02 20:24:12 +01:00
aa5927d8a1
findCUDNN
2019-11-06 14:04:23 +01:00
2594f59d0d
conv2d ok, but deconv ha different dim with tensorrt
2019-10-30 16:45:46 +01:00
f247300469
test simple
2019-10-30 15:40:48 +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
d6d93a74f8
fix
2019-10-30 09:42:54 +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
e9ec582223
Shortcat ok
2019-10-29 15:08:44 +01:00
058f9b795f
resnet weight export fix
2019-10-28 19:20:02 +01:00
4ac2c87d1c
cmake fix
2019-10-28 18:39:28 +01:00
42a1ea02b9
resnet merge
2019-10-28 18:38:23 +01:00
92f3d1c548
fixed install cmake
2019-10-01 18:47:31 +02:00
bbc4dda635
removed buildtype
2019-09-17 17:16:46 +02:00
xavier
Davide Sapienza
luca
Francesco Gatti
fbagni
mbosi