if ((0 != CV_MAT_DEPTH(typePrev)) || (0 != CV_MAT_DEPTH(typeNext)))
return false;
- if (_prevPts.empty() || _prevPts.size().height != 1 || _prevPts.type() != CV_32FC2)
+ if (_prevPts.empty() || _prevPts.type() != CV_32FC2 || (!_prevPts.isContinuous()))
return false;
+ if ((1 != _prevPts.size().height) && (1 != _prevPts.size().width))
+ return false;
+ size_t npoints = _prevPts.total();
bool useInitialFlow = (0 != (flags & OPTFLOW_USE_INITIAL_FLOW));
if (useInitialFlow)
{
- if (_nextPts.size() != _prevPts.size() || _nextPts.type() != CV_32FC2)
+ if (_nextPts.empty() || _nextPts.type() != CV_32FC2 || (!_prevPts.isContinuous()))
+ return false;
+ if ((1 != _nextPts.size().height) && (1 != _nextPts.size().width))
+ return false;
+ if (_nextPts.total() != npoints)
return false;
}
+ else
+ {
+ _nextPts.create(_prevPts.size(), _prevPts.type());
+ }
PyrLKOpticalFlow opticalFlow;
opticalFlow.winSize = winSize;
UMat umatErr;
if (_err.needed())
{
- _err.create(_prevPts.size(), CV_32FC1);
+ _err.create((int)npoints, 1, CV_32FC1);
umatErr = _err.getUMat();
}
else
- umatErr.create(_prevPts.size(), CV_32FC1);
+ umatErr.create((int)npoints, 1, CV_32FC1);
- _nextPts.create(_prevPts.size(), _prevPts.type());
- _status.create(_prevPts.size(), CV_8UC1);
+ _status.create((int)npoints, 1, CV_8UC1);
UMat umatNextPts = _nextPts.getUMat();
UMat umatStatus = _status.getUMat();
return opticalFlow.sparse(_prevImg.getUMat(), _nextImg.getUMat(), _prevPts.getUMat(), umatNextPts, umatStatus, umatErr);
OCL_TEST_P(PyrLKOpticalFlow, Mat)
{
- cv::Mat frame0 = readImage("optflow/rubberwhale1.png", cv::IMREAD_GRAYSCALE);
+ static const int npoints = 1000;
+ static const float eps = 0.03f;
+
+ cv::Mat frame0 = readImage("optflow/RubberWhale1.png", cv::IMREAD_GRAYSCALE);
ASSERT_FALSE(frame0.empty());
UMat umatFrame0; frame0.copyTo(umatFrame0);
- cv::Mat frame1 = readImage("optflow/rubberwhale1.png", cv::IMREAD_GRAYSCALE);
+ cv::Mat frame1 = readImage("optflow/RubberWhale2.png", cv::IMREAD_GRAYSCALE);
ASSERT_FALSE(frame1.empty());
UMat umatFrame1; frame1.copyTo(umatFrame1);
std::vector<cv::Point2f> pts;
- cv::goodFeaturesToTrack(frame0, pts, 1000, 0.01, 0.0);
+ cv::goodFeaturesToTrack(frame0, pts, npoints, 0.01, 0.0);
std::vector<cv::Point2f> cpuNextPts;
std::vector<unsigned char> cpuStatusCPU;
UMat umatNextPts, umatStatus, umatErr;
OCL_ON(cv::calcOpticalFlowPyrLK(umatFrame0, umatFrame1, pts, umatNextPts, umatStatus, umatErr, winSize, maxLevel, criteria, flags, minEigThreshold));
- std::vector<cv::Point2f> nextPts(umatNextPts.cols); umatNextPts.copyTo(nextPts);
- std::vector<unsigned char> status; umatStatus.copyTo(status);
- std::vector<float> err; umatErr.copyTo(err);
+ std::vector<cv::Point2f> nextPts; umatNextPts.reshape(2, 1).copyTo(nextPts);
+ std::vector<unsigned char> status; umatStatus.reshape(1, 1).copyTo(status);
+ std::vector<float> err; umatErr.reshape(1, 1).copyTo(err);
ASSERT_EQ(cpuNextPts.size(), nextPts.size());
ASSERT_EQ(cpuStatusCPU.size(), status.size());
double bad_ratio = static_cast<double>(mistmatch) / (nextPts.size());
- ASSERT_LE(bad_ratio, 0.02f);
+ ASSERT_LE(bad_ratio, eps);
}
OCL_INSTANTIATE_TEST_CASE_P(Video, PyrLKOpticalFlow,