yolov3 yolo_cpp_dll 调用

1.编译

1.1 如出现 cudnn宏问题，查看cudnn版本是否正确

1.2 包含目录

..\..\3rdparty\include;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include;%(AdditionalIncludeDirectories);$(CudaToolkitIncludeDir);$(cudnn)\include

1.3包含库目录

$(CUDA_PATH)lib\$(PlatformName);$(cudnn)\lib\x64;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64;%(AdditionalLibraryDirectories)

1.4包含库

..\..\3rdparty\lib\x64\pthreadVC2.lib
cublas.lib
curand.lib
cudart.lib

可以发现，主要为cuda以及cudnn的依赖

2.test

#include <iostream>
#include "opencv2/core/core.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/opencv.hpp"
#include "Yolov2Detector.h"
using namespace std;
vector<string> objName = { "person", "Saftyhelmetperson" };
const string CFG_FILE = "\\cfg\\helmet02.cfg";
const string WEIGHTS_FILE = "\\helmet02_59400.weights";
#define ORG_PIC_WIDTH MAX_PIC_WIDTH
#define ORG_PIC_HEIGHT MAX_PIC_HEIGHT

#define MAX_PIC_WIDTH 1920
#define MAX_PIC_HEIGHT 1080
int main()
{
   string curPath = "E:/work/yolov3test/yolov3test/";
   string cfgPath = curPath + CFG_FILE;
   string weightsPath = curPath + WEIGHTS_FILE;
   Yolov2Detector *m_pDetector = new Yolov2Detector(cfgPath, weightsPath);
   Mat img(ORG_PIC_WIDTH, ORG_PIC_HEIGHT, CV_8UC3);
   char cPath[MAX_PATH] = "E:/work/yolov3test/yolov3test/11.jpg" ;
   img =cv::imread(cPath);
   int iDetectObjNums = 0; //报警目标个数
   float m_fProbThres = 0.25;
   int iAllObjCount = 0; //检测到的所有目标个数
   string obj_name = "Saftyhelmetperson";//需要报警的目标，不含该目标将直接删除服务器发来的路径对应的图片
   vector <string>::iterator sElement = std::find(objName.begin(),
       objName.end(), obj_name);

auto result = m_pDetector->Detect(img, m_fProbThres);
result = m_pDetector->TrackingID(result);

iAllObjCount = result.size();
std::cout << "检测到 " << iAllObjCount << "个目标" << std::endl;

   int iPosition = distance(objName.begin(), sElement); //报警目标在vector中的position
   for (int i = 0; i < iAllObjCount; i++)
   {
       if (result.at(i).obj_id == iPosition)
       {
           iDetectObjNums += 1;
       }
   }
   m_pDetector->DrawBoxes(img, result, objName);
   cv::imshow("",img);
   cv::waitKey(0);
}