Commit Graph

168 Commits

Author SHA1 Message Date
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