mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
218 Commits 11 Branches 5 Tags
75d5e20843291b16e876972f6488dd3b3cb2d592
Commit Graph

218 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Micaela Verucchi 75d5e20843 Add writing results on file for map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-10 14:35:25 +01:00
Micaela Verucchi 0773f216b3 Update README 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-08 18:12:08 +01:00
Davide Sapienza b734baca0c Update README 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-03 17:34:01 +01:00
Davide Sapienza 99e809737f Move multiple demo files into one 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-03 17:30:35 +01:00
Davide Sapienza c94159a2bf Add draw method to Yolo3Detection class. 
Update yolo demo

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-03 17:08:25 +01:00
Micaela Verucchi 1d38d88c85 Fix dependencies problems 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-02-27 16:48:10 +01:00
xavier 3be254b08f Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-26 17:42:20 +01:00
xavier c36c898948 Fix include error 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-21 11:14:52 +01:00
xavier c4cef1aa02 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-21 10:49:35 +01:00
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 200e9466f6 Move pre-processing on GPU 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-14 18:44:13 +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