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 */