main.cpp

#include <iostream>
#include "VM_RTSPServer.h"
#include "VM_RTSPToImage.h"
#include "VM_YOLO.h"
#include "VM_ThreadSafeList.h"

void rtsp_to_image_thread(std::vector<VM_Buffer> rtspURLs) {
    VM_RTSPToImage r2i(rtspURLs);
    r2i.Init();
    r2i.Invoke();
}
int main(){
    // std::cout<<"start VideoMonitorServer..."<<std::endl;
    // VM_RTSPServer server;
    // server.Init();
    // server.Invoke();
    // std::cout<<"press anykey to stop..."<<std::endl;
    // getchar();
    // server.Stop();
    // std::cout<<"test rtsp stream to images..."<<std::endl;
    std::vector<VM_Buffer> rtspURLs;
    //rtspURLs.push_back("rtsp://stream.strba.sk:1935/strba/VYHLAD_JAZERO.stream");
    rtspURLs.push_back("rtsp://stream.strba.sk:1935/strba/VYHLAD_JAZERO.stream");
    rtspURLs.push_back("rtsp://user:1234.abcd@192.168.10.95/Streaming/Channels/101");
    rtspURLs.push_back("rtsp://user:1234.abcd@192.168.10.95/Streaming/Channels/102");
    // VM_RTSPToImage r2i(rtspURLs);
    // r2i.Init();
    // r2i.Invoke();
    std::thread rtspThread(rtsp_to_image_thread, rtspURLs);

    std::string onnx_file_path = std::string(PROJECT_ROOT_DIR) + "/models/yolov11n.onnx";
    std::string trt_file_path = std::string(PROJECT_ROOT_DIR) + "/models/yolov11n_8.4.3.1.trt";
    //std::string input_image_path = std::string(PROJECT_ROOT_DIR) + "/sources/images/bus.jpg";
    //std::string output_image_path = std::string(PROJECT_ROOT_DIR) + "/sources/images/out_bus.jpg";
    int batch_size = 1;
    int channel = 3;
    int width = 640;
    int height = 640;
    YOLO yolo(onnx_file_path,trt_file_path,batch_size,channel,width,height);
    //cv::Mat img = cv::imread(input_image_path);
    //yolo.Inference(img,output_image_path);
    VM_RTSPToImage::StructDeliveryData data;
    cv::Mat img;
    int count=1;
    std::string output_image_path = std::string(PROJECT_ROOT_DIR) + "/sources/SaveFrame/yolo/";
    while(1)
    {
        if(VM_RTSPToImage::delivery_list.empty())
        {
            usleep(100);
            continue;
        }
        std::cout<<"count: "<<count<<std::endl;
        VM_RTSPToImage::delivery_list.pop(data);
        int batch_size = 1;
        int channel = 3;
        int width = data.width_gpu_srcImage;
        int height = data.height_gpu_srcImage;
        img = data.srcImage;
        //std::cout<<"rtsp: "<<data.rtsp_path<<std::endl;
        yolo.Inference(img,output_image_path + std::to_string(count)+".jpg");
        count++;
    }
    std::cout<<"over~~"<<std::endl;
    rtspThread.join();
    return 0;
}