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>
This commit is contained in:
Micaela Verucchi
2020-03-17 10:35:02 +01:00
parent 9c25d15ff2
commit 1d388ca51a
5 changed files with 637 additions and 46 deletions
+1 -1
View File
@@ -44,7 +44,7 @@ int main(int argc, char *argv[]) {
cnet.init(net);
break;
case 'm':
mbnet.init(net);
mbnet.init(net, 512, 81);
break;
default:
FatalError("Network type not allowed (3rd parameter)\n");