 Francesco Gatti
|
5e71b99265
|
YoloRT save bias, mask and clasesName into RT file
|
2022-03-30 20:46:51 +02:00 |
|
 perseusdg
|
2ffe07057e
|
Mnist works at the moment with trt8,others like yolo4tiny and mobilenet generate the engine files but crash after throwing nvifer1::CudaRuntimeError and when demo is being run ,it doesnt deserialize properly and crashes
|
2021-08-29 03:18:58 +05:30 |
|
 Micaela Verucchi
|
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 |
|
|
Micaela Verucchi
|
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 |
|
|
Micaela Verucchi
|
533bb48789
|
Add json detection creation for codalab check
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
|
2020-05-11 11:57:58 +02:00 |
|
|
Micaela Verucchi
|
af13e7c954
|
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
|
3eda9b9219
|
Add GPU version for yolo3detection
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
|
2020-03-23 21:29:29 +01:00 |
|
|
Micaela Verucchi
|
df37e11709
|
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
|
bbcc33c0cf
|
Refactoring for detection NN
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
|
2020-03-20 21:14:12 +01:00 |
|
|
Francesco Gatti
|
f6527f51e3
|
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
|
4b85a2238a
|
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
|
296a6cbc87
|
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
|
fe206ea24c
|
Fix dependencies problems
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
|
2020-02-27 16:48:10 +01:00 |
|
|
Davide Sapienza
|
c68d6f318e
|
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
|
2020-01-20 16:41:19 +01:00 |
|
|
xavier
|
da4f246157
|
add DLA, plugin for shortcut and leaky. new verison 0.4
|
2020-01-15 21:48:18 +01:00 |
|
|
nvidia
|
854a4c316a
|
Add ResizeLayerRT plugin
Signed-off-by: nvidia <micaelaverucchi@gmail.com>
|
2019-12-20 15:27:41 +01:00 |
|
|
Francesco Gatti
|
ca62784f57
|
include dir fix, cmake dir
|
2019-09-17 15:22:39 +02:00 |
|