Commit Graph

8 Commits

Author SHA1 Message Date
Micaela Verucchi 2fbac7705d Move download of weights inside build folder
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-17 22:36:25 +02:00
Micaela Verucchi fabad7bace 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
Micaela Verucchi 3502c5b676 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
Davide Sapienza ce20868bad 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
Davide Sapienza df888a3457 Add CenterNet based on Resnet101, TensorRT not implemented.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2019-12-18 18:59:54 +01:00
Davide Sapienza 02936fa928 ResNet working 2019-10-29 17:21:31 +01:00
Davide Sapienza e9ec582223 Shortcat ok 2019-10-29 15:08:44 +01:00
Micaela Verucchi 42a1ea02b9 resnet merge 2019-10-28 18:38:23 +01:00