Peter M. Groen / openbr

Commit 771540b2c6e7d51305fdf647a929e715dde4a925

Authored by Austin Van Blanton
1 parent 4cb891c4

Cvt to gray before OF, propagate file from src to dst

Inline Side-by-side
Showing 1 changed file with 8 additions and 6 deletions
openbr/plugins/opticalflow.cpp
  View file @771540b
1 1 #include <opencv2/video/tracking.hpp>
2 2 #include "openbr_internal.h"
  3 +#include "openbr/core/opencvutils.h"
3 4  
4 5 using namespace cv;
5 6  
... ... @@ -11,7 +12,7 @@ namespace br
11 12 * \brief Gets a one-channel dense optical flow from two images
12 13 * \author Austin Blanton \cite imaus10
13 14 */
14   -class OpticalFlowTransform : public UntrainableTransform
  15 +class OpticalFlowTransform : public UntrainableMetaTransform
15 16 {
16 17 Q_OBJECT
17 18 Q_PROPERTY(double pyr_scale READ get_pyr_scale WRITE set_pyr_scale RESET reset_pyr_scale STORED false)
... ... @@ -33,18 +34,19 @@ class OpticalFlowTransform : public UntrainableTransform
33 34 void project(const Template &src, Template &dst) const
34 35 {
35 36 // get the two images put there by AggregateFrames
36   - Mat prevImg = src[0];
37   - Mat nextImg = src[1];
38   - Mat flow;
  37 + if (src.size() < 2) return;
  38 + Mat prevImg = src[0], nextImg = src[1], flow, flowOneCh;
  39 + if (src[0].channels() != 1) OpenCVUtils::cvtGray(src[0], prevImg);
  40 + if (src[1].channels() != 1) OpenCVUtils::cvtGray(src[1], nextImg);
39 41 calcOpticalFlowFarneback(prevImg, nextImg, flow, pyr_scale, levels, winsize, iterations, poly_n, poly_sigma, flags);
40   -
  42 +
41 43 // the result is two channels
42 44 std::vector<Mat> channels(2);
43 45 split(flow, channels);
44   - Mat flowOneCh;
45 46 magnitude(channels[0], channels[1], flowOneCh);
46 47  
47 48 dst += flowOneCh;
  49 + dst.file = src.file;
48 50 }
49 51 };
50 52  
... ...