Skip to content
Snippets Groups Projects

CUDA optical flow smoothing

Merged Sebastian Hahta requested to merge feature/off-cuda into master
5 files
+ 84
2
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -15,6 +15,7 @@ public:
@@ -15,6 +15,7 @@ public:
OFDisparityFilter() : n_max_(0), threshold_(0.0), size_(0, 0) {} // TODO: invalid state
OFDisparityFilter() : n_max_(0), threshold_(0.0), size_(0, 0) {} // TODO: invalid state
OFDisparityFilter(cv::Size size, int n_frames, float threshold);
OFDisparityFilter(cv::Size size, int n_frames, float threshold);
void filter(cv::Mat &disp, const cv::Mat &rgb);
void filter(cv::Mat &disp, const cv::Mat &rgb);
 
void filter(cv::cuda::GpuMat &disp, const cv::cuda::GpuMat &optflow);
private:
private:
int n_;
int n_;
@@ -23,6 +24,7 @@ private:
@@ -23,6 +24,7 @@ private:
cv::Size size_;
cv::Size size_;
cv::Mat disp_;
cv::Mat disp_;
 
cv::cuda::GpuMat disp_old_;
cv::Mat gray_;
cv::Mat gray_;
cv::Mat flowxy_;
cv::Mat flowxy_;
Loading