Function main

internal/bin/classification.cpp:375–472

}

int main(int argc, char* argv[])
{
    // params
    int repeat_count = DEFAULT_REPEAT_CNT;

    std::string model_format;
    std::string proto_file;
    std::string model_file;
    std::string label_file;
    std::string image_file;
    int model_fnum = 0;

    int img_h = 224;
    int img_w = 224;
    float scale = 1.f;
    float mean[3] = {104.007, 116.669, 122.679};

    std::vector<int> hw;
    std::vector<float> ms;
    int res;

    if(argc == 1)
    {
        std::cout << "[Usage]: " << argv[0] << " [-h]\n"
                  << "    [-f model_format] [-p proto_file] [-m model_file]\n"
                  << "    [-l label_file] [-i image_file] [-g img_h,img_w]\n"
                  << "    [-s scale] [-w mean[0],mean[1],mean[2]] [-r repeat_count]\n";
        return 0;
    }
    while((res = getopt(argc, argv, "f:p:m:l:i:g:s:w:r:h")) != -1)
    {
        switch(res)
        {
            case 'f':
                model_format = optarg;
                break;
            case 'p':
                proto_file = optarg;
                break;
            case 'm':
                model_file = optarg;
                break;
            case 'l':
                label_file = optarg;
                break;
            case 'i':
                image_file = optarg;
                break;
            case 'g':
                hw = ParseString<int>(optarg);
                if(hw.size() != 2)
                {
                    std::cerr << "Error -g parameter.\n";
                    return -1;
                }
                img_h = hw[0];
                img_w = hw[1];
                break;