Function main

yolox_example/main.cpp:417–446

}

int main(int argc, char** argv) {
    cv::CommandLineParser parser(argc, argv, keys);
    if (argc < 3) {
        help(argv);
        return -1;
    }
    const std::string model_path = parser.get<cv::String>("@model");
    std::string image_filename = cv::samples::findFile(parser.get<cv::String>("input"));
    std::string outout_filename = parser.get<cv::String>("output");

    cv::Mat img = cv::imread(image_filename);
    if (img.empty()) {
        std::cout << "Couldn't open the image " << image_filename
                  << ". Usage: inpaint <image_name>\n"
                  << std::endl;
        return 0;
    }
    std::vector<float> input(3 * 640 * 640, 0);
    preprocess(img, input.data());
    std::vector<float> output_data = inference(model_path.c_str(), input);
    float* predict_ptr = output_data.data();
    int img_w = img.cols;
    int img_h = img.rows;
    float scale = std::min(INPUT_W / (img.cols * 1.0), INPUT_H / (img.rows * 1.0));
    std::vector<OutputInfo> output_infos;

    postpreprocess(predict_ptr, output_infos, scale, img_w, img_h);
    draw_output_infos(img, output_infos, outout_filename);
    return 0;
}