eyemNNDetector.h
612 字节
#pragma once
//
// eyemNNDetectorͷ
//
#ifndef __EYEMNNDETECTOR_H
#define __EYEMNNDETECTOR_H
#include "eyemLib.h"
#include "yoloWrapper.h"
class NNDetector {
public:
NNDetector(const std::string& detector_config_path,
const std::string& detector_model_path, const cv::Size& input_size);
NNDetector() {};
//ü
void setInputParams(float fConfidence, float fNMSThreshold);
std::vector<cv::Rect> detect(cv::InputArray img);
protected:
class Impl;
cv::Ptr<Impl> p;
};
cv::Ptr<NNDetector> pNNDetector;
#ifdef _DEBUG
//cv::Ptr<YoloDarknet> pClassifier;
#endif
#endif/* __EYEMNNDETECTOR_H */