#pragma once
#include "IDetection.h"

class ElectronPlateDectect :IDetection
{

public:
	IDetection::DectectResult GetStateResult(cv::Mat img, cv::Rect rec);
	IDetection::DectectResult GetDigitResult(cv::Mat img, cv::Rect rec);
	vector<IDetection::DectectResult> GetDigitResults(cv::Mat img, cv::Rect rec);

	bool Detect(cv::Mat& SrcImg);
	bool Init(bool isCuda);
private:
	IDetection::DectectResult resultValue;
	vector<IDetection::DectectResult> resultValues;
	vector<Output> output;
	cv::dnn::Net net;

	/*const std::vector<std::string> className = {
		"three_rows_digit_voltage",
		"three_rows_digit_current",
		"two_rows_digit_voltage_current",
		"one_rows_digit_temperature",
		"one_rows_digit_count",
		"one_rows_digit_voltage",
		"one_rows_digit_current",
	};*/
	const std::vector<std::string> className = { "digit","state" };
	
};