flowcalculatorcuda.h

#ifndef FLOWCACLULATORCUDA_H
#define FLOWCACLULATORCUDA_H
 
#include "flowcalculator.h"
 
#include <opencv2/cudaoptflow.hpp>
#include <opencv2/cudaarithm.hpp>
#include <opencv2/cudawarping.hpp>
#include <opencv2/cudaimgproc.hpp>
 
class FlowCalculatorCuda : public FlowCalculator
{
public:
    FlowCalculatorCuda(int numLevels=5, double pyrScale=0.5, bool fastPyramids=false, int winSize=13, int numIters=10, int polyN=5, double polySigma=1.1, int flags=0);
    ~FlowCalculatorCuda();
    double calculateFlow(cv::Mat fromMat, cv::Mat toMat) override;
 
private:
    cv::Ptr<cv::cuda::FarnebackOpticalFlow> m_farn;
};
 
#endif // FLOWCACLULATORCUDA_H