Function main

tools/cpp/generateIO.cpp:39–177

}

int main(int argc, char* argv[]) {
    if (argc < 4) {
        MNN_PRINT("Usage: ./generateLlmIO model inputJson outputDir externalFilePath\n");
        return 1;
    }

    std::string ExternalFilePath;
    std::string modelPath = std::string(argv[1]);
    std::string inputJson = argv[2];
    std::string outputDir = argv[3];
    if (argc >= 5) {
        ExternalFilePath = argv[4];
    }

    rapidjson::Document document;
    std::ifstream fileNames(inputJson.c_str());
    std::ostringstream output;
    output << fileNames.rdbuf();
    auto outputStr = output.str();
    document.Parse(outputStr.c_str());
    if (document.HasParseError()) {
        MNN_ERROR("Invalid json\n");
        return 0;
    }
    int shapeIndex = 0;
    std::shared_ptr<MNN::Express::Module> net;

    if (document.HasMember("configs")) {
        if (!(MNNCreateDir(outputDir.c_str()))) {
            MNN_PRINT("Failed to create dir %s.\n", outputDir.c_str());
        }
        auto configsArray = document["configs"].GetArray();
        for (auto& configObj : configsArray) {
            std::map<std::string, float> inputInfo;
            std::map<std::string, std::string> inputType;
            std::vector<std::string> inputNames;
            std::vector<std::string> outputNames;
            std::map<std::string, std::vector<int>> inputShape;
            std::vector<MNN::Express::VARP> inputs;
            std::vector<MNN::Express::VARP> outputs;
            if (configObj.HasMember("inputs")) {
                auto inputsInfo = configObj["inputs"].GetArray();
                for (auto iter = inputsInfo.begin(); iter != inputsInfo.end(); iter++) {
                    auto obj = iter->GetObject();
                    std::string type = "float";
                    std::string name = obj["name"].GetString();
                    inputNames.emplace_back(name);
                    if (obj.HasMember("type")) {
                        type = obj["type"].GetString();
                        inputType.insert(std::make_pair(name, type));
                    }
                    if (obj.HasMember("value")) {
                        float value;
                        if (type == "int") {
                            value = (float)obj["value"].GetInt();
                        } else {
                            value = obj["value"].GetFloat();
                        }