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

61 Commits

Author SHA1 Message Date
Micaela Verucchi f2991bec5c Add variable batch for detector update 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 09:32:27 +02:00
Micaela Verucchi 360adbd1a1 Add verbose define 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-15 09:41:27 +02:00
Micaela Verucchi 7df266cac4 Adapt detection classes to use batches, adapt demos, update README 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-14 16:41:51 +02:00
Micaela Verucchi 7e8dbc8296 Add flag to disable visualization and save video 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 15:08:10 +02:00
Micaela Verucchi 5eb0cbfcef Add json detection creation for codalab check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 11:57:58 +02:00
Francesco Gatti 585f66e2f6 warning fix 2020-04-09 20:05:10 +02:00
Francesco Gatti c248a94101 warning fix 2020-04-09 01:00:14 +02:00
Micaela Verucchi 6a08118e3d Refactoring & documentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 11:20:48 +02:00
Micaela Verucchi 6e55e2376f Move extraction of name into function in utils 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 10:09:15 +02:00
Micaela Verucchi 065c4e2b58 Minor fix on output file name in map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 09:35:54 +02:00
Micaela Verucchi 733755e7e0 Add evaluation to tk::dnn namespace, style fix also 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-06 18:12:38 +02:00
Micaela Verucchi 43567dc3ea Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors. 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-01 19:18:43 +02: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 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 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 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
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 5c347b1e11 Fix print on times.csv 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-12 20:18:02 +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
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 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
xavier 9c4ea524fc Add script to download COCO val2017 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 14:56:54 +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 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
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
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 ef18be7b7b Add mAP computation and demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-05 18:39:13 +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
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
xavier da4f246157 add DLA, plugin for shortcut and leaky. new verison 0.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
xavier 57d7743f7e Change opencv funcion call (due to OpenCV 4) 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-01-15 09:55:10 +01:00
nvidia 854a4c316a Add ResizeLayerRT plugin 
Signed-off-by: nvidia <micaelaverucchi@gmail.com>
 2019-12-20 15:27:41 +01:00
Francesco Gatti ec02c7292f save layer names in rt file 2019-09-16 19:41:59 +02:00
Francesco Gatti 77f031c0f4 save video result 2019-09-16 10:35:29 +02:00
mbosi a038e966d9 yolo3 flir ok 2019-09-15 16:19:30 +02:00
mbosi eef1fd321f added label to demo bounding box visualization 2019-05-02 14:30:50 +02:00
rcavicchioli 851c6a366c arg fix 2019-02-19 11:10:34 +01:00
Francesco Gatti de04ae1cab doc 2019-02-19 09:03:33 +00:00
Francesco Gatti 87fe342ca2 yoloRT load anchors 2019-02-18 21:39:14 +01:00
Francesco Gatti bdd8e0bc26 yolo3plug fix 2019-02-18 18:55:48 +00:00
Francesco Gatti 13063b904d yolo3 ok 2019-02-18 15:51:57 +00:00
Francesco Gatti 0d682136de yolo3 berkeley ok 2019-02-18 15:37:39 +00:00