#pragma once
#include "IDetection.h"

class DisconnectorStateDectect :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 = {
		"on",
		"off"
	};
};