| 126 | } |
| 127 | |
| 128 | static void _printHelp() { |
| 129 | std::cout << "Usage: " << " model_to_benchmark [loop_count] [forwardtype] [numberThread]" << std::endl; |
| 130 | std::cout << "model_to_benchmark: " << std::endl; |
| 131 | std::cout << "\t default: run standard models" << std::endl; |
| 132 | std::cout << "\t MobileNetV1_{numClass}_{width}_{resolution}, width: {1.0, 0.75, 0.5, 0.25}, resolution: {224, 192, 160, 128}, e.g: MobileNetV1_100_1.0_224" << std::endl; |
| 133 | std::cout << "\t MobileNetV2_{numClass}, e.g: MobileNetV2_100" << std::endl; |
| 134 | std::cout << "\t ResNet_{numClass}_{layer}, layer: {18, 34, 50, 101, 152}, e.g: ResNet_100_18" << std::endl; |
| 135 | std::cout << "\t GoogLeNet_{numClass}, e.g: GoogLeNet_100" << std::endl; |
| 136 | std::cout << "\t SqueezeNet_{numClass}, e.g: SqueezeNet_100" << std::endl; |
| 137 | std::cout << "\t ShuffleNet_{numClass}_{group}, group: [1, 2, 3, 4, 8], e.g: ShuffleNet_100_4" << std::endl; |
| 138 | } |
| 139 | static std::vector<std::string> gDefaultModels = { |
| 140 | "MobileNetV1_1000_1.0_224", |
| 141 | "MobileNetV2_1000", |