be5864748a
Use yaml config file for the demo instead of param list
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-23 16:29:40 +01:00
1de804f98d
Code cleanup and readme fixes
2021-04-09 13:17:41 +05:30
2e92944f1d
Opencv cuda fix
2021-03-31 03:44:51 -07:00
e94e1f7622
tkdnn first patch for windows
2021-03-24 22:26:46 +05:30
6aa8666be5
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
2d4dececb6
Builds on windows successfully,issues with deserialization and downloading weights
2021-01-26 20:17:43 +04:00
f778e1aa99
Fixed boxes to float, add conf thresh as param
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-05 19:55:10 +02:00
377310af50
Add variable batch for detector update
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 09:32:27 +02:00
23b40de508
Add verbose define
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-15 09:41:27 +02:00
40456592fc
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
2e3cb52cff
Add flag to disable visualization and save video
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 15:08:10 +02:00
ed0596a52c
warning fix
2020-04-09 01:00:14 +02:00
bbcc33c0cf
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
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
01a42ffe27
Move multiple demo files into one
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:30:35 +01:00
7f239efdc0
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
c68d6f318e
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-01-20 16:41:19 +01:00
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
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
da4f246157
add DLA, plugin for shortcut and leaky. new verison 0.4
2020-01-15 21:48:18 +01:00
0f0c8c28e1
Merge with master, works with Jetpack 4.3
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-01-15 19:24:19 +01:00
57d7743f7e
Change opencv funcion call (due to OpenCV 4)
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-01-15 09:55:10 +01:00
854a4c316a
Add ResizeLayerRT plugin
...
Signed-off-by: nvidia <micaelaverucchi@gmail.com >
2019-12-20 15:27:41 +01:00
ec02c7292f
save layer names in rt file
2019-09-16 19:41:59 +02:00
77f031c0f4
save video result
2019-09-16 10:35:29 +02:00
a038e966d9
yolo3 flir ok
2019-09-15 16:19:30 +02:00
eef1fd321f
added label to demo bounding box visualization
2019-05-02 14:30:50 +02:00
851c6a366c
arg fix
2019-02-19 11:10:34 +01:00
de04ae1cab
doc
2019-02-19 09:03:33 +00:00
87fe342ca2
yoloRT load anchors
2019-02-18 21:39:14 +01:00
bdd8e0bc26
yolo3plug fix
2019-02-18 18:55:48 +00:00
13063b904d
yolo3 ok
2019-02-18 15:51:57 +00:00
0d682136de
yolo3 berkeley ok
2019-02-18 15:37:39 +00:00
Micaela Verucchi
perseusdg
Francesco Gatti
Davide Sapienza
mbosi
rcavicchioli