ImageRecognition/MeterDectect.h

#pragma once
#include "IDetection.h"

class MeterDectect :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 = {
         "circular_arrester_current",
        "atmospheric_pressure",
        "digital_gear",
        "rect_arrester_current",
        "quarter_square_ammeter",
        "double_pointer_count",
        "transformers_oil_surface_thermometer",
        "close",
        "open",
        "round_single_point_oil_level",
        "oil_thermometer",
        "old_oil_thermometer",
	};*/
    const vector<std::string> className = { 
        "atmospheric_pressure", 
        "ampere_meter",
        "transformers_oil_surface_thermometer",
        "rect_arrester_current",
        "double_pointer_count",
        "round_single_point_oil_level",
        "voltage_meter",
        "oil_thermometer",
        "open",
        "close" };
};