ContactDectect.h 444 B

1234567891011121314151617181920212223
  1. #pragma once
  2. #include "IDetection.h"
  3. class ContactDectect :IDetection
  4. {
  5. public:
  6. IDetection::DectectResult GetStateResult(cv::Mat img, cv::Rect rec);
  7. IDetection::DectectResult GetDigitResult(cv::Mat img, cv::Rect rec);
  8. bool Detect(cv::Mat& SrcImg);
  9. bool Init(bool isCuda);
  10. private:
  11. IDetection::DectectResult resultValue;
  12. cv::dnn::Net net;
  13. const std::vector<std::string> className = {
  14. "connection",
  15. "disconnection",
  16. };
  17. };