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

239 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
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
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