camera_bounding_box_detector.h

#pragma once
// Camera object bounding box detection frame processor class header
#include "camera_bounding_box_factory.h"
 
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/video/background_segm.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/features2d.hpp>
#include <opencv2/videoio.hpp>
 
#include <cstring>
#include <iostream>
#include <vector>
#include <list>
 
using namespace cv;
using namespace std;
 
const int kParamMinArea = 100;
 
const double kparamBackgroundLearnRate = -1; //0.001;
 
class CameraBoundingBoxDetector : public CameraBoundingBoxDetectorInterface
{
public:
 
    CameraBoundingBoxDetector();
    
    CameraBoundingBoxDetector(int history, int threshold, bool shadows);
 
    bool DetectBoundingBox(BYTE* input_frame_buffer,int length, int image_width, int image_height, int image_type, vector<BoundingBoxPoint> &bounding_rect) override;
    
    void SetBackgroundFrame(BYTE* background_frame_buffer, int length, int image_width, int image_height, int image_type) override;
 
    BackgroundFilterType background_type = BackgroundFilterType::kMixed;
 
    int param_min_area = kParamMinArea;
 
    double param_background_learn_rate = kparamBackgroundLearnRate;
 
private:
    void SetBackground(Mat background_frame);
 
    bool DetectBoundingBoxPosition(Mat input_frame, cv::Rect &bounding_rect);
    
    Mat saved_background_frame_;
    Mat gray_;
    Mat img_output_;
    Mat diff_frame_;
    Mat diff_frame1_;
    Mat diff_frame2_;
    Mat output_frame_;
    Mat output_frame1_;
    Mat output_frame2_;
    
    Ptr<BackgroundSubtractor> bg_proc_;
};