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

26 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
Francesco Gatti c248a94101 warning fix 2020-04-09 01:00:14 +02: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 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
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 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