mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
229 Commits 11 Branches 5 Tags
32dd807ab05a20356dafb2b059dc5b452f9fdfd6
Commit Graph

229 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
xavier 32dd807ab0 Add Yolov3_tiny512 test cpp 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-13 12:16:18 +01:00
xavier 9d17cb42f4 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 20:18:11 +01:00
xavier db10ebeb34 Fix print on times.csv 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-12 20:18:02 +01:00
Francesco Gatti 62e7882744 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 18:43:33 +01:00
Francesco Gatti 13b4dffd71 csresnext50-panet-spp works with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-12 18:24:41 +01:00
Francesco Gatti f6527f51e3 Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-11 18:37:26 +01:00
Micaela Verucchi 66e3b98b04 Update README.md 2020-03-11 12:04:28 +01:00
Micaela Verucchi 0cc2666e26 Update README.md 2020-03-11 11:51:28 +01:00
xavier d0d462c015 Merge 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:43:26 +01:00
xavier 2d46f16b70 Add tiny yolo 512 test 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:40:11 +01:00
Micaela Verucchi e6aa73d7bc Update README.md 2020-03-10 15:26:57 +01:00
Micaela Verucchi 4b85a2238a 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 7ea31d123a Update README 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-08 18:12:08 +01:00
Davide Sapienza 7dbbbe4d9a Update README 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-03 17:34:01 +01:00
Davide Sapienza 01a42ffe27 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 296a6cbc87 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 fe206ea24c Fix dependencies problems 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-02-27 16:48:10 +01:00
xavier 3eb079dd13 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 110bf56dc4 Fix include error 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-21 11:14:52 +01:00
xavier 209a3e8492 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-21 10:49:35 +01:00
xavier 38a1b9dcb2 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 2c1df5619f Move pre-processing on GPU 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-14 18:44:13 +01:00
Davide Sapienza 97b88ef52d 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 4c2d2a7965 Update README.md 2020-02-11 15:12:27 +01:00
xavier 636d899f9f Merge README 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 14:58:25 +01:00
xavier 7e21b10aee Add script to download COCO val2017 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 14:56:54 +01:00
Micaela Verucchi 6e2ff405f4 Update README.md 2020-02-11 11:28:20 +01:00
xavier 1936e54870 Change README.md, add config.yaml, refactoring 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 11:25:42 +01:00
xavier b5c0baa99d Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-11 10:32:52 +01:00
xavier 801b8b5641 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 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
xavier 10831ab450 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-10 18:19:12 +01:00
xavier d601e980f6 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 b06dc286e9 Update README.md 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 11:19:39 +01:00
Davide Sapienza c02238ddc8 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 e72aa348a0 Add DLA34 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:56:33 +01:00
Davide Sapienza 62fe82ce9e Update ResNet101 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:54:04 +01:00
Davide Sapienza 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
xavier 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
xavier 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
xavier 5c501f529b Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-07 09:06:29 +01:00
Davide Sapienza 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
xavier c21a0687ce Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-06 18:24:49 +01:00
Davide Sapienza 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
xavier 37019597af Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-05 18:39:32 +01:00
xavier d8eeb36d4b Add mAP computation and demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-05 18:39:13 +01:00
Davide Sapienza 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
Davide Sapienza 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
Davide Sapienza fe85c26888 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 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