mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
245 Commits 11 Branches 5 Tags
7c55dcb708fbd1c7a4a776503d5b948cce4db2fe
Commit Graph

245 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Micaela Verucchi 7c55dcb708 Add tp tests for yolo3512 and yolo3tiny512 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-26 15:06:49 +01:00
Micaela Verucchi 3eda9b9219 Add GPU version for yolo3detection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 21:29:29 +01:00
Micaela Verucchi df37e11709 Modify map demo, using abstract class. Move draw function in abstract class 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 19:32:37 +01:00
Micaela Verucchi bbcc33c0cf Refactoring for detection NN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 21:14:12 +01:00
Micaela Verucchi c7d9c38ea0 Add OPENCV_CUDA define, to allow having preprocess both in CPU and GPU (Mobilenet and Centernet) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 10:21:11 +01:00
Micaela Verucchi 58fe723b6a Style fix, useless params removed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-19 18:13:35 +01:00
Micaela Verucchi f44f377771 Add preprocess function, allow preprocess on GPU for mobilenetdetection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-19 17:53:29 +01:00
Micaela Verucchi 41b1135fb3 Add support for mobilenet in map_demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 15:08:43 +01:00
Micaela Verucchi 43fd92f701 Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 13:18:05 +01:00
Micaela Verucchi 1d388ca51a Add mobilenetv2ssdlite512 test. 
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 10:35:02 +01:00
Micaela Verucchi 9c25d15ff2 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00
Micaela Verucchi 82d907971c Fix download weights yolov3 versions 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-14 18:20:32 +01:00
Francesco Gatti 555b32b5dc Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti cf7fbadcd2 Add function to download weigths if do no exist, for each test. Add some controls in map demo. 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:02:07 +01:00
xavier 5e0c1879bc Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-13 12:17:21 +01:00
xavier 32dd807ab0 Add Yolov3_tiny512 test cpp 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-13 12:16:18 +01:00
Micaela Verucchi 8aa792925d Update README.md 2020-03-12 20:42:44 +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