test_lite_yolov8.cpp

//
// Created by ai-test1 on 24-7-8.
//

#include "lite/lite.h"



static void test_tensorrt()
{
#ifdef ENABLE_TENSORRT
    std::string engine_path = "../../..//examples/hub/trt/yolov8n_fp32.engine";
    std::string test_img_path = "../../..//examples/lite/resources/test_lite_yolov5_1.jpg";
    std::string save_img_path = "../../..//examples/logs/test_lite_yolov8_trt_1.jpg";

    lite::trt::cv::detection::YOLOV8 *yolov8  = new lite::trt::cv::detection::YOLOV8(engine_path);

    cv::Mat test_image = cv::imread(test_img_path);

    std::vector<lite::types::Boxf> detected_boxes;

    yolov8->detect(test_image,detected_boxes,0.5f,0.4f);

    std::cout<<"trt yolov8 detect done!"<<std::endl;
    lite::utils::draw_boxes_inplace(test_image, detected_boxes);
    cv::imwrite(save_img_path, test_image);

    delete yolov8;
#endif
}

static void test_lite()
{
    test_tensorrt();
}



int main(__unused int argc, __unused char *argv[])
{
    test_lite();
    return 0;
}