m

Scheaven

2021-08-12 d91bfe2d12d0fb5daf2be0a7587045d85a50c4a1

m

 CMakeLists.txt

@@ -9,7 +9,7 @@
set(CUDA_Path /usr/local/cuda)
# find_package(OpenCV REQUIRED PATHS "/opt/toolkits/opencv")
set(OPENCV_DIR /opt/toolkits/opencv)
set(TensorRT_DIR /opt/toolkits/TensorRT7)
set(TensorRT_DIR /opt/toolkits/TensorRT-7.2.3.4)
set(LOG_DIR /opt/toolkits/log4cplus)

include_directories(${OPENCV_DIR}/include/opencv4)

 config.json

@@ -1,10 +1,9 @@
{

  "so_file_path": "/opt/vasystem/libs/Detect/libdemo.so",

  "runtime": "/opt/vasystem/libs/Detect:/usr/local/cuda-11.1/lib64:",  // 项目所用到的环境

  "runtime": "/opt/vasystem/libs/Detect:/usr/local/cuda-11.1/lib64:",

  "param": {

    // "model_path": "/opt/vasystem/bin/models/baseDetector/baseDetetor.bin" // para里边自己算法可能用到的参数

    "model_path": "/opt/vasystem/bin/models/baseDetector/baseDetetor_small.bin", // para里边自己算法可能用到的参数

    "type":2,

    "model_path": "/opt/vasystem/bin/models/baseDetector/baseDetetor.bin",

    "type":1,

    "max_cam_num": 8,

    "wander_time": 5,

    "mv_velocity": 10.0,


 demo.cpp

@@ -32,7 +32,7 @@
    TImage *img = new TImage();
    TImage *img2 = new TImage();
    int i = 0;
    cap = VideoCapture("/data/disk2/01_Scheaven/data/cam_53.avi");
    cap = VideoCapture("../1.mp4");

    while(1){        /* code */
        cap >> frame;

 src/core/ari_manager.cpp

@@ -141,6 +141,7 @@
    // cv::imshow("img",image0);

    // cv::waitKey(0);

    this->detector->detect(batch_img, batch_res);

    Timer::getInstance()->out("eveTime detect");



    t_result->targets = (Target*)malloc(sizeof(Target)*batch_res[0].size());

    int w_count = 0;

@@ -202,8 +203,9 @@
        }

    }

    std::cout << "eve batch_res size:: "<< batch_res[0].size() << " w_count: " << w_count <<std::endl;

    t_result->count = w_count;

    // draw_SDK_result(cam_id, frame, t_result);

    t_result.count = w_count;

    Timer::getInstance()->out("eveTime draw_SDK_result");

}






 src/h_interface.cpp

@@ -18,8 +18,8 @@


API void* create(const char *conf, int *max_chan)

{

    // CLog::Initialize("/opt/vasystem/bin/models/baseDetector/log.properties");

    CLog::Initialize("../config/log4cplus.properties");

    CLog::Initialize("/opt/vasystem/bin/models/baseDetector/log.properties");

    // CLog::Initialize("../config/log4cplus.properties");

    ReadJsonFromFile(conf);

    AriManager *handle = new AriManager();

    *max_chan = m_staticStruct::max_cam_num;


 src/utils/draw_util.cpp

@@ -80,8 +80,8 @@
    cv::waitKey(1);

}



cv::VideoWriter writer("./01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);

// cv::VideoWriter writer("/opt/vasystem/valog/01_Scheanve/01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);

//cv::VideoWriter writer("./01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);

cv::VideoWriter writer("/opt/vasystem/valog/01_Scheanve/01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);



void draw_SDK_result(const int cam_id, cv::Mat mat_img, TResult* t_result)

{