Function main

tools/bin/test_accuracy.cpp:163–314

}

int main(int argc, char* argv[])
{
    // Load image file
    std::vector<std::string> images;
    LoadImageFile(images, image_file);
    // Load Label file
    std::vector<int> labels;
    LoadLableFile(labels, label_file);

    /*
     1. prototxt
     2. mobile
     3. mean
     4. scale
     5. input tensor
     6. output tensor
     7. input weight and input height
     8. image_path
    */
    if(argc < 9)
    {
        std::cout << "The params are not enough \n";
    }

    std::string text_file = argv[1];
    std::string model_file = argv[2];
    std::string mean_val = argv[3];
    std::string scale_val = argv[4];
    std::string input_tensor_name = argv[5];
    std::string output_tensor_name = argv[6];
    std::string image_wh = argv[7];
    std::string val_file_path = argv[8];

    float means[3];
    float scales[3];
    string delima = ",";

    std::vector<string> res0;
    std::vector<string> res1;
    std::vector<string> wh;

    split(mean_val, delima, &res0);
    split(scale_val, delima, &res1);
    split(image_wh, delima, &wh);

    for(int i = 0; i < 3; i++)
    {
        means[i] = atof(res0[i].c_str());
        scales[i] = atof(res1[i].c_str());
    }

    int img_w = atoi(wh[0].c_str());
    int img_h = atoi(wh[1].c_str());
    int img_size = img_h * img_w * 3;
    float* input_data = ( float* )malloc(sizeof(float) * img_size);

    init_tengine();