All Downloads are FREE. Search and download functionalities are using the official Maven repository.

srcnativelibs.Include.OpenCV.opencv2.contrib.detection_based_tracker.hpp Maven / Gradle / Ivy

#pragma once

#if defined(__linux__) || defined(LINUX) || defined(__APPLE__) || defined(ANDROID)

#include 
#include 

#include 

class DetectionBasedTracker
{
    public:
        struct Parameters
        {
            int minObjectSize;
            int maxObjectSize;
            double scaleFactor;
            int maxTrackLifetime;
            int minNeighbors;
            int minDetectionPeriod; //the minimal time between run of the big object detector (on the whole frame) in ms (1000 mean 1 sec), default=0

            Parameters();
        };

        DetectionBasedTracker(const std::string& cascadeFilename, const Parameters& params);
        virtual ~DetectionBasedTracker();

        virtual bool run();
        virtual void stop();
        virtual void resetTracking();

        virtual void process(const cv::Mat& imageGray);

        bool setParameters(const Parameters& params);
        const Parameters& getParameters();


        typedef std::pair Object;
        virtual void getObjects(std::vector& result) const;
        virtual void getObjects(std::vector& result) const;

    protected:
        class SeparateDetectionWork;
        cv::Ptr separateDetectionWork;
        friend void* workcycleObjectDetectorFunction(void* p);


        struct InnerParameters
        {
            int numLastPositionsToTrack;
            int numStepsToWaitBeforeFirstShow;
            int numStepsToTrackWithoutDetectingIfObjectHasNotBeenShown;
            int numStepsToShowWithoutDetecting;

            float coeffTrackingWindowSize;
            float coeffObjectSizeToTrack;
            float coeffObjectSpeedUsingInPrediction;

            InnerParameters();
        };
        Parameters parameters;
        InnerParameters innerParameters;

        struct TrackedObject
        {
            typedef std::vector PositionsVector;

            PositionsVector lastPositions;

            int numDetectedFrames;
            int numFramesNotDetected;
            int id;

            TrackedObject(const cv::Rect& rect):numDetectedFrames(1), numFramesNotDetected(0)
            {
                lastPositions.push_back(rect);
                id=getNextId();
            };

            static int getNextId()
            {
                static int _id=0;
                return _id++;
            }
        };

        int numTrackedSteps;
        std::vector trackedObjects;

        std::vector weightsPositionsSmoothing;
        std::vector weightsSizesSmoothing;

        cv::CascadeClassifier cascadeForTracking;


        void updateTrackedObjects(const std::vector& detectedObjects);
        cv::Rect calcTrackedObjectPositionToShow(int i) const;
        void detectInRegion(const cv::Mat& img, const cv::Rect& r, std::vector& detectedObjectsInRegions);
};

namespace cv
{
    using ::DetectionBasedTracker;
} //end of cv namespace

#endif