#pragma once
#include "IDetection.h"

class CharacterDectect :IDetection
{

public:
	IDetection::DectectResult GetStateResult(cv::Mat img, cv::Rect rec);
	IDetection::DectectResult GetDigitResult(cv::Mat img, cv::Rect rec);
	vector<IDetection::DectectResult> GetStateResults();
	vector < IDetection::DectectResult> GetDigitResults();
	bool Detect(cv::Mat& SrcImg);
	bool Init(bool isCuda);
private:
	IDetection::DectectResult resultValue;
	vector <IDetection::DectectResult> resultStateValues;
	vector <IDetection::DectectResult> resultDigitValues;
	cv::dnn::Net net;
	
	const std::vector<std::string> className = {
		                "on",
					    "off",
						"stored_energy",
						"stored_energy_graphics",
						"green_on",
						"red_on"
	};
};