mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
208 Commits 11 Branches 5 Tags
808a84131c78d493382af8db3b5ec9a754f2db31
Commit Graph

208 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
xavier 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
Davide Sapienza 40a4a55cd0 Fix a bug in the deformable kernel. 
There was a wrong variable initialization.

Fixes: 51ffcb1f50 ("Optimize deformable kernel")

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-11 16:28:17 +01:00
Micaela Verucchi 217b3fbfb5 Update README.md 2020-02-11 15:12:27 +01:00
xavier b1a6b45f5f Merge README 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 14:58:25 +01:00
xavier 9c4ea524fc Add script to download COCO val2017 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 14:56:54 +01:00
Micaela Verucchi 21535dd6b3 Update README.md 2020-02-11 11:28:20 +01:00
xavier d0026ff235 Change README.md, add config.yaml, refactoring 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 11:25:42 +01:00
xavier e01d7c6b13 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-11 10:32:52 +01:00
xavier 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
Davide Sapienza 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
xavier 3ef2cc65f5 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-10 18:19:12 +01:00
xavier c7cb3e7571 Add avg precision, recall, f1score computation, other minors 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-10 18:19:08 +01:00
Davide Sapienza 34ecb6a1bb Update README.md 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 11:19:39 +01:00
Davide Sapienza 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
Davide Sapienza 932de41e38 Add DLA34 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:56:33 +01:00
Davide Sapienza 5d07e98a9c Update ResNet101 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:54:04 +01:00
Davide Sapienza 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
xavier 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
xavier 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
xavier 002355e7a7 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-07 09:06:29 +01:00
Davide Sapienza 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
xavier 0a4c75f834 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-06 18:24:49 +01:00
Davide Sapienza 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
xavier dc130fc834 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-05 18:39:32 +01:00
xavier ef18be7b7b Add mAP computation and demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-05 18:39:13 +01:00
Davide Sapienza 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
Davide Sapienza 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
Davide Sapienza 04918f539e Implement DLA34, CUDNN and TensorRT work. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-01-24 19:30:32 +01:00
Davide Sapienza 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
Davide Sapienza b7a6e8c8f5 Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-01-20 16:41:19 +01:00
luca 6a3b261bc9 compile with tensorrt 5 2020-01-20 14:51:38 +01:00
Davide Sapienza 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
Davide Sapienza 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
Francesco Gatti c3aadb089b Update README.md 2020-01-16 18:24:35 +01:00
Francesco Gatti 07eae338fe Update README.md 2020-01-16 18:21:34 +01:00
xavier da4f246157 add DLA, plugin for shortcut and leaky. new verison 0.4 v0.4 2020-01-15 21:48:18 +01:00
xavier 0f0c8c28e1 Merge with master, works with Jetpack 4.3 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-01-15 19:24:19 +01:00
Francesco Gatti f3f5daf3db Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-01-15 18:07:44 +01:00
Francesco Gatti c2d73623e5 support clion 2020-01-15 18:07:40 +01:00
xavier c32a0be257 Batchnorm eps fix, works on jetpack 4.3 2020-01-15 18:06:02 +01:00
xavier 57d7743f7e Change opencv funcion call (due to OpenCV 4) 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-01-15 09:55:10 +01:00
fbagni 23a1365dc4 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2019-12-23 16:14:24 +01:00
fbagni 7e7f480e2b Yolov3_tiny works on tensorRT 
Signed-off-by: fbagni <gattinomicino>
 2019-12-23 16:11:41 +01:00
Davide Sapienza 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
nvidia f0ea2c027f Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2019-12-20 15:27:55 +01:00
nvidia 854a4c316a Add ResizeLayerRT plugin 
Signed-off-by: nvidia <micaelaverucchi@gmail.com>
 2019-12-20 15:27:41 +01:00
Davide Sapienza 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
Davide Sapienza 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
Davide Sapienza 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
mbosi 6bf9179acc fix to drivework global path 2019-12-12 12:30:24 +01:00