24 lines
550 B
C
24 lines
550 B
C
|
#pragma once
|
|||
|
|
|
|||
|
|
// CYoloV4
|
|||
|
|
#include "yolo_v2_class.hpp"
|
|||
|
|
|
|||
|
|
class CYoloV4
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
CYoloV4();
|
|||
|
|
virtual ~CYoloV4();
|
|||
|
|
void InitYoloV4();
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
void DrawObjectBoxes(cv::Mat mat_img, std::vector<bbox_t> result_vec, std::vector<std::string> obj_names, int current_det_fps=-1, int current_cap_fps=-1);
|
|||
|
|
void TargetDetection(int nCameraIdx, cv::Mat &mat_img, vector<CRect> &findRects);
|
|||
|
|
std::vector<std::string> ObjectsNamesFromFile(CString strFilename);
|
|||
|
|
std::vector<std::string> m_vObjects_Names; //<2F><><EFBFBD><EFBFBD>Ŀ<EFBFBD><C4BF><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
|
Detector *m_pDetector;
|
|||
|
|
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
|