#pragma once
#include "IDetection.h"

class OilLevelDectect :IDetection
{

public:
	IDetection::DectectResult GetStateResult(cv::Mat img, cv::Rect rec);
	IDetection::DectectResult GetDigitResult(cv::Mat img, cv::Rect rec);

	bool Detect(cv::Mat& SrcImg);
	bool Init(bool isCuda);
private:
	IDetection::DectectResult resultValue;
	cv::dnn::Net net;

	const std::vector<std::string> className = {
				"buoy_header",
			   "buoy_body",
	};
};