Function main

samples/cpp/run_model/classification.cpp:223–249

}

int main(int argc, char* argv[]) {
    if (argc != 3) {
        fprintf(stderr, "Usage: %s <image_file> <onnx_model_file>\n", argv[0]);
        return -1;
    }

    const char* img_path = argv[1];
    Mat src_img = imread(img_path);
    if (src_img.empty()) {
        fprintf(stderr, "read image file %s failed!\n", img_path);
        return -1;
    }

    const bool resize_input = false; // pplnn can adapt dynamic input size even if onnx model has static input size
    if (resize_input) {
        resize(src_img, src_img, Size(224, 224));
    }

    const char* onnx_model_path = argv[2];
    int32_t ret = RunClassificationModel(src_img, onnx_model_path);
    if (ret != 0) {
        fprintf(stderr, "run classification model failed!\n");
        return -1;
    }

    return 0;
}