Function main

benchmark/benchmarkExprModels.cpp:149–232

};

int main(int argc, const char* argv[]) {
    std::cout << "MNN Expr Models benchmark" << std::endl;
    size_t loop = 10;
    MNNForwardType forward = MNN_FORWARD_CPU;
    size_t numThread = 4;
    if (argc <= 1) {
        _printHelp();
        return 0;
    }
    if (((argc > 1) && (strcmp(argv[1], "help") == 0)) || argc > 5) {
        _printHelp();
        return 0;
    }
    std::vector<std::string> models;
    if (((argc > 1) && (strcmp(argv[1], "default") == 0)) || argc > 5) {
        models = gDefaultModels;
    } else {
        models = {argv[1]};
    }

    if (argc >= 3) {
        loop = atoi(argv[2]);
    }
    if (argc >= 4) {
        forward = static_cast<MNNForwardType>(atoi(argv[3]));
    }
    if (argc >= 5) {
        numThread = atoi(argv[4]);
    }
    std::cout << "Forward type: **" << forwardType(forward) << "** thread=" << numThread << std::endl;
    ScheduleConfig config;
    config.type = forward;
    config.numThread = numThread;
    BackendConfig bnConfig;
    bnConfig.precision = BackendConfig::Precision_Low;
    bnConfig.power = BackendConfig::Power_High;
    config.backendConfig = &bnConfig;

    std::vector<float> costs;

    // ResNet18 benchmark
    for (auto model : models) {
        auto modelArgs = splitArgs(model.c_str(), "_");
        auto modelType = modelArgs[0];
        int numClass   = atoi(modelArgs[1].c_str());
        if (modelType == "MobileNetV1") {
            auto mobileNetWidthType = EnumMobileNetWidthTypeByString(modelArgs[2]);
            if (mobileNetWidthType < 0) {
                std::cout << "Not support MobileNetWidthType " << modelArgs[2] << std::endl;
                std::cout << "Only [1.0, 0.75, 0.5, 0.25] be support" << std::endl;
                return 1;
            }
            auto mobileNetResolutionType = EnumMobileNetResolutionTypeByString(modelArgs[3]);
            if (mobileNetResolutionType < 0) {
                std::cout << "Not support MobileNetResolutionType " << modelArgs[3] << std::endl;
                std::cout << "Only [224, 192, 160, 128] be support" << std::endl;
                return 1;
            }