#pragma once
//
//	yoloWrapperͷ
//
#ifndef __YOLOWRAPPER_H
#define __YOLOWRAPPER_H

#include "opencv2/dnn.hpp"
#include "opencv2/imgproc.hpp"


class YoloWrapper
{
public:
	YoloWrapper() {};
	~YoloWrapper() { };
	int init(const std::string& config_path, const std::string& model_path);
	std::vector<cv::Rect> forward(cv::Mat img);
private:
	cv::dnn::Net net_;
};

#endif/* __YOLOWRAPPER_H */