Function GetModelConfig

test/test.cc:260–305

    }

    ModelConfig GetModelConfig() {
        ModelConfig config;
        config.model_type = ConvertModelType(FLAGS_mt);
        if (config.model_type == MODEL_TYPE_TNN || config.model_type == MODEL_TYPE_OPENVINO ||
            config.model_type == MODEL_TYPE_NCNN) {
            std::string network_path = FLAGS_mp;
            int size                 = static_cast<int>(network_path.size());
            std::string model_path;

            // TNN file names: xxx.tnnproto  xxx.tnnmodel
            // NCNN file names: xxx.param xxx.bin
            if (config.model_type == MODEL_TYPE_TNN) {
                model_path = network_path.substr(0, size - 5) + "model";
            } else if (config.model_type == MODEL_TYPE_NCNN) {
                model_path = network_path.substr(0, size - 5) + "bin";
            } else {
                model_path = network_path.substr(0, size - 3) + "bin";
            }

            std::ifstream proto_stream(network_path);
            if (!proto_stream.is_open() || !proto_stream.good()) {
                printf("read proto_file failed!\n");
                return config;
            }
            auto buffer =
                    std::string((std::istreambuf_iterator<char>(proto_stream)), std::istreambuf_iterator<char>());
            config.params.push_back(buffer);

            if (config.model_type == MODEL_TYPE_TNN || config.model_type == MODEL_TYPE_NCNN) {
                std::ifstream model_stream(model_path, std::ios::binary);
                if (!model_stream.is_open() || !model_stream.good()) {
                    config.params.push_back("");
                    return config;
                }
                std::stringstream model_content;
                model_content << model_stream.rdbuf();

                config.params.push_back(model_content.str());
            } else {
                config.params.push_back(model_path);
            }
        } else {
            config.params.push_back(FLAGS_mp);
        }
        return config;
    }

    NetworkConfig GetNetworkConfig() {
        NetworkConfig config;