mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
309 Commits 11 Branches 5 Tags
70227b6502f2ee77a853eefd7d52cc462a841e6b
Commit Graph

309 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Micaela Verucchi 8d5bd653de Update README.md 2020-03-29 20:03:59 +02:00
Davide Sapienza e217a0e83f Fix opencv cuda include error 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 10:43:49 +01:00
Davide Sapienza e540213da6 Add the Int8 calibrator and the tensorRT Int8 inference 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 00:48:14 +01:00
Micaela Verucchi f4b976c793 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 785a3d8dd5 Add GPU version for yolo3detection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 21:29:29 +01:00
Micaela Verucchi 3bc648c07b 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 8b9516da97 Refactoring for detection NN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 21:14:12 +01:00
Micaela Verucchi 997e857e32 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 5c6d6ae0af Style fix, useless params removed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-19 18:13:35 +01:00
Micaela Verucchi cae0b84687 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 f6129311d6 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 1a50065bde 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 14d850751d 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 4a2a244162 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00
Francesco Gatti a7c3761960 eigen assert fix 2020-03-14 19:50:55 +01:00
Micaela Verucchi 94971fbdcb Fix download weights yolov3 versions 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-14 18:20:32 +01:00
Francesco Gatti 677b6b2698 Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti 21b8146232 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 01995abdf8 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-13 12:17:21 +01:00
xavier 51b9780baf Add Yolov3_tiny512 test cpp 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-13 12:16:18 +01:00
Micaela Verucchi 35bbda0ab1 Update README.md 2020-03-12 20:42:44 +01:00
xavier a65c8aa91e Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 20:18:11 +01:00
xavier 5c347b1e11 Fix print on times.csv 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-12 20:18:02 +01:00
Francesco Gatti ef1ebea5e9 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 18:43:33 +01:00
Francesco Gatti 4845edbc88 csresnext50-panet-spp works with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-12 18:24:41 +01:00
Francesco Gatti 5f4508a5a6 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 963b2c40d1 Update README.md 2020-03-11 12:04:28 +01:00
Micaela Verucchi d5722cae3f Update README.md 2020-03-11 11:51:28 +01:00
xavier 98770ee0c5 Merge 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:43:26 +01:00
xavier 07a61b0e37 Add tiny yolo 512 test 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:40:11 +01:00
Micaela Verucchi 0c5e31476e Update README.md 2020-03-10 15:26:57 +01:00
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
Francesco Gatti 2d378e9070 imuodom fix 2020-03-09 22:05:34 +01:00
Francesco Gatti 6fee8d5ec5 - ImuOdom model into class 
- Route layer input array hard copy
- fix utils
 2020-03-09 19:44:21 +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
Francesco Gatti cd1e1c1582 LSTM ok 2020-02-16 17:21:58 +01:00
Francesco Gatti 4b48a2f38c structure ok, result wrong 2020-02-16 17:08:19 +01:00
Francesco Gatti 0ac292ea48 works but it need cleaning 2020-02-16 16:28:39 +01:00
Francesco Gatti 8736a8c3da LSTM params 2020-02-15 20:37:08 +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
Francesco Gatti f5706d8523 lstm return seq 2020-02-13 23:21:28 +01:00
Francesco Gatti 0329318ca3 removed unused var 2020-02-13 23:10:48 +01:00