Commit Graph

6 Commits

Author SHA1 Message Date
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
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 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
Micaela Verucchi 14d850751d 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