mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
256 Commits 11 Branches 5 Tags
e4900120dbd2d1c76f2e7eff47dee36114ff4e29
Commit Graph

5 Commits

Author SHA1 Message Date
Micaela Verucchi 79a05a8159 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 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
Micaela Verucchi f44f377771 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 43fd92f701 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 1d388ca51a 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