派生自 Algorithm/baseDetector

Scheaven
2021-06-03 7e5f7f3f140b048d97f751b2e148065eb039a572
add m
2个文件已修改
29 ■■■■■ 已修改文件
CMakeLists.txt 16 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
main.cpp 13 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
CMakeLists.txt
@@ -71,10 +71,9 @@
endif()
# set(OPENCV_DIR /data/disk2/opt/01_opencv/opencv3.4.8)
set(OPENCV_DIR /data/disk2/opt/01_opencv/opencv2.4.13)
set(OPENCV_DIR /opt/toolkits/opencv)
link_directories(/usr/local/cuda-10.0/lib64)
link_directories(/usr/local/cuda/lib64)
#link_directories(/usr/local/lib)
SET(LIBS
@@ -82,9 +81,9 @@
        nvToolsExt
        opencv_core
        opencv_highgui
        # opencv_imgcodecs
        opencv_imgcodecs
        opencv_imgproc
        # opencv_videoio
        opencv_videoio
        opencv_video
        opencv_calib3d
        opencv_features2d
@@ -101,7 +100,7 @@
#aux_source_directory(./src DIR_SRCS)  # 搜索当前目录下的所有.cpp文件
#add_library(demo ${SRC_LIST})   #明确指定包含哪些源文件
include_directories(/usr/local/cuda-10.0/include)
include_directories(/usr/local/cuda/include)
include_directories(".")
include_directories("./lib")
include_directories("./lib/core")
@@ -111,14 +110,15 @@
#include_directories("/usr/include/json/")
#include_directories("/usr/include")
#include_directories("/usr/local/include/")
include_directories(${OPENCV_DIR}/include)
include_directories(${OPENCV_DIR}/include/opencv4)
#设置链接库搜索目录,项目中lib路径
link_directories(${PYTORCH_DIR}/lib)  #动态链接库目录/bazel-bin/tensorflow
#link_directories(/usr/local/lib)
link_directories(${OPENCV_DIR}/lib)
#link_directories(/usr/lib/x86_64-linux-gnu/)
link_directories(/home/scheaven/anaconda3/lib/)
link_directories(/home/basic/anaconda3/lib/)
#添加要编译的可执行文件
main.cpp
@@ -5,14 +5,8 @@
#include "std_target.h"
#include "time.h"
#ifdef OPENCV
#include <opencv2/opencv.hpp>            // C++
#pragma comment(lib, "opencv_core249.lib")
#pragma comment(lib, "opencv_imgproc249.lib")
#pragma comment(lib, "opencv_highgui249.lib")
#endif    // OPENCV
using namespace std;
using namespace cv;
void* handle;
@@ -39,7 +33,7 @@
    TImage *img = new TImage();
    TImage *img2 = new TImage();
    int i = 0;
    cap = VideoCapture("/data/disk2/01_dataset/03_human/34983128.mp4");
    cap = VideoCapture("/data/disk2/01_Scheaven/data/cam_53.avi");
    // string path="/data/disk1/workspace/07_CPP/build/re";
    // std::vector<string> video_path_vec;
@@ -48,7 +42,8 @@
    // for (auto& file:video_path_vec)
    // {
    while(1){        /* code */
    while(1)
    {        /* code */
        // printf("----start--%s\n", file);
        // cout<<"----------------\n\n"<<file<<endl;
        // frame = imread(file);
@@ -68,7 +63,6 @@
            cout << "=======t_result->count==" << t_result->count  << endl;
            for (int i=0; i<t_result->count; i++)
                cout << "====1111111111===confidence:" << t_result->targets[i].confidence << endl;
            // if (t_result->count>0)
            // {
            //    imwrite("/data/disk2/01_dataset/03_human/02_person/positive/"+file.substr(file.find_last_of("/")), frame);
@@ -76,7 +70,6 @@
            // {
            //     imwrite("/data/disk2/01_dataset/03_human/02_person/negate/"+file.substr(file.find_last_of("/")), frame);
            // }
        }else{
            cout << "-----------------------over--" << endl;