thirdparty/yolov7/TensorRT-For-YOLO-Series-cuda-python/cpp/norm/main.cpp

#pragma once
#include <string>
#include "yolo.hpp"

int main(int argc, char** argv) {
    std::string engine_file_path = "";
    if (argc == 4 && std::string(argv[2]) == "-i") 
    {
        engine_file_path = argv[1];
    } 
    else {
        std::cerr << "arguments not right!" << std::endl;
        std::cerr << "./yolo ../model_trt.engine -i ../*.jpg  // deserialize file and run inference" << std::endl;
        return -1;
    }

    const std::string input_image_path {argv[3]};
    YOLO yolo(engine_file_path);
    yolo.detect_img(input_image_path);
    return 0;
}