1234567891011121314151617181920212223242526272829 |
- #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"
- };
- };
|