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; //¼ì²âÄ¿±êÃû³Æ
|
|
Detector *m_pDetector;
|
|
|
|
};
|
|
|
|
|