PersonBehaviorDetect.h 550 B

123456789101112131415161718192021222324
  1. #pragma once
  2. #include "IDetection.h"
  3. class PersonBehaviorDetect :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. "smoke",
  15. "drink",
  16. "fall"
  17. };
  18. };
  19. #pragma once