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
cd1e1c1582
LSTM ok
2020-02-16 17:21:58 +01:00
4b48a2f38c
structure ok, result wrong
2020-02-16 17:08:19 +01:00
0ac292ea48
works but it need cleaning
2020-02-16 16:28:39 +01:00
8736a8c3da
LSTM params
2020-02-15 20:37:08 +01:00
200e9466f6
Move pre-processing on GPU
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-14 18:44:13 +01:00
f5706d8523
lstm return seq
2020-02-13 23:21:28 +01:00
0329318ca3
removed unused var
2020-02-13 23:10:48 +01:00
e8280d3a65
LSTM to be tested
2020-02-13 23:04:29 +01:00
c876fa05ae
LSTM cudnn test
2020-02-13 19:27:18 +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
217b3fbfb5
Update README.md
2020-02-11 15:12:27 +01:00
b1a6b45f5f
Merge README
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 14:58:25 +01:00
9c4ea524fc
Add script to download COCO val2017
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 14:56:54 +01:00
21535dd6b3
Update README.md
2020-02-11 11:28:20 +01:00
d0026ff235
Change README.md, add config.yaml, refactoring
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 11:25:42 +01:00
e01d7c6b13
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-11 10:32:52 +01:00
73548b179e
Read parameters for mAP from yaml, add yampl-cpp dependency
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 10:32:42 +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
3ef2cc65f5
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-10 18:19:12 +01:00
c7cb3e7571
Add avg precision, recall, f1score computation, other minors
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-10 18:19:08 +01:00
34ecb6a1bb
Update README.md
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 11:19:39 +01:00
da8ba7d8ce
Add Anaconda environment for ResNet101 and DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:58:04 +01:00
932de41e38
Add DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:56:33 +01:00
5d07e98a9c
Update ResNet101 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:54:04 +01:00
17a083ae28
Remove mallocs and frees from the kernels
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-07 14:50:05 +01:00
da24c2e752
Refactoring map_demo, add evaluation.h and evaluation.cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 12:47:02 +01:00
9bc4ba0eff
Add mAP 0.5:0.95, other small fix
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 09:09:40 +01:00
002355e7a7
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-07 09:06:29 +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
0a4c75f834
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-06 18:24:49 +01:00
516c1e6636
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
dc130fc834
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-05 18:39:32 +01:00
ef18be7b7b
Add mAP computation and demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-05 18:39:13 +01:00
507d30bf52
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
0d6acaec41
Add grouped convolutions in CUDNN and tensorRT.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-05 14:32:58 +01:00
04918f539e
Implement DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-24 19:30:32 +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
b7a6e8c8f5
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-01-20 16:41:19 +01:00
6a3b261bc9
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
c3aadb089b
Update README.md
2020-01-16 18:24:35 +01:00
07eae338fe
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
xavier
Francesco Gatti
Davide Sapienza
luca