Francesco Gatti
af9bb411be
darknet parser tested
2020-06-01 14:55:08 +02:00
Francesco Gatti
21518c5eef
darknet parse all net to be tested
2020-06-01 12:22:55 +02:00
Micaela Verucchi
702e38a150
Add names files
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:42:06 +02:00
Micaela Verucchi
fe07c8fb26
Read classes' names from file
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:34:48 +02:00
Francesco Gatti
4b7e80ae0d
yolo3 parsed ok
2020-05-30 22:11:00 +02:00
Francesco Gatti
27a83e79d3
darknet parser to be tested on yolo3
2020-05-30 19:38:26 +02:00
Micaela Verucchi
3e701e7ba3
Add fields to darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:44:17 +02:00
Francesco Gatti
664a717353
parse yolo layers
2020-05-30 18:44:02 +02:00
Micaela Verucchi
26e6ba52a5
Add some fields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:00:22 +02:00
Francesco Gatti
1755e734b1
parse layer and network
2020-05-30 17:59:00 +02:00
Micaela Verucchi
98173ec15c
Add darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:57:48 +02:00
Francesco Gatti
179e07e830
parse line by line
2020-05-30 17:27:22 +02:00
Francesco Gatti
f9d28c58ca
darknet parser interface
2020-05-30 16:58:40 +02:00
Francesco Gatti
54074780cf
filter yaw ok
2020-05-03 19:12:10 +02:00
Francesco Gatti
b34c013ad7
merge
2020-05-03 15:58:00 +02:00
Francesco Gatti
099aa1accd
Imu odom weights
2020-05-03 15:56:30 +02:00
Francesco Gatti
eb1d30d1b1
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-05-03 15:54:42 +02:00
Francesco Gatti
8d45259b15
SEP model
2020-05-03 15:54:39 +02:00
Francesco Gatti
72dfd92c0a
imu odom SEP model
2020-05-03 15:53:35 +02:00
Micaela Verucchi
3be4fe0117
Modify download
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:40:35 +02:00
Micaela Verucchi
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
Francesco Gatti
1d284bf7d6
Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree
2020-04-28 14:54:18 +02:00
Micaela Verucchi
fdfda38fbf
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +02:00
Micaela Verucchi
dbe54f4b67
Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 14:34:12 +02:00
Micaela Verucchi
a761fef159
Add bdd new tests, add BDD100K_val download, fix yolo3-512 link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-24 12:39:57 +02:00
Francesco Gatti
7597539076
merge tkDNN
2020-04-24 11:18:22 +02:00
Francesco Gatti
2a6f7db4af
Flatten batch to be checked
2020-04-23 01:18:27 +02:00
Francesco Gatti
1cbf18c21e
test batch
2020-04-23 00:54:03 +02:00
Francesco Gatti
9655c29b94
RouteRt is not used
2020-04-21 19:50:45 +00:00
Francesco Gatti
a91b2f46ba
batch seems ok in yolo3_berkely
...
layers to be checked:
DeformableConvRT
FlattenConcatRT
ReshapeRT
RouteRT (dont know why but seems working)
2020-04-21 19:41:40 +00:00
Francesco Gatti
ef91439c87
batch size > 1
2020-04-21 19:20:44 +02:00
Micaela Verucchi
ec7620ebb6
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
Francesco Gatti
70227b6502
warning fix
2020-04-14 14:26:06 +02:00
Davide Sapienza
50ada3db60
Remove the optional parameters from the Pooling layer.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 19:20:27 +02:00
Micaela Verucchi
a8fd4cabf3
Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 19:02:25 +02:00
Davide Sapienza
8a575f6447
Remove the final parameter from the layers
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 18:48:03 +02:00
Micaela Verucchi
af761786ea
Fix pooling, add return code in each test, add return code handling in test_all_tests script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 15:25:05 +02:00
Francesco Gatti
2389df8626
Merge branch 'master' into cnet
2020-04-09 00:57:43 +02:00
Francesco Gatti
73fb12be0f
gnuplot imuodom
2020-04-08 16:53:35 +02:00
Micaela Verucchi
18aff44c99
Add imuodom download and test in script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:50:40 +02:00
Micaela Verucchi
8c1eb69494
Merge with master works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 14:59:42 +02:00
Micaela Verucchi
e036c5da34
Update download url
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 12:34:02 +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
9a1b673db8
Modify weights download for some nets, now also calibration tables are downloaded in tests folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 16:22:23 +02:00
Micaela Verucchi
9e08f4125a
Modify weights download link for mobilenetv2ssd512 and yolo3_tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-02 09:41:24 +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
Davide Sapienza
0474f019cf
Add TKDNN_MODE variable to the name of the network.
...
This commit permits to obtain different .rt files for different
precision optimizations of the same network.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 15:08:25 +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
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
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