mirror of
https://github.com/opencv/opencv.git
synced 2024-11-26 12:10:49 +08:00
3bfc69740b
Merged pull requests: #890 from caorong:patch-1 #893 from jet47:gpu-arm-fixes #933 from pengx17:2.4_macfix_cont #935 from pengx17:2.4_filter2d_fix #936 from bitwangyaoyao:2.4_perf #937 from bitwangyaoyao:2.4_fixPyrLK #938 from pengx17:2.4_surf_sample #939 from pengx17:2.4_getDevice #940 from SpecLad:autolock #941 from apavlenko:signed_char #946 from bitwangyaoyao:2.4_samples2 #947 from jet47:fix-gpu-arm-build #948 from jet47:cuda-5.5-support #952 from SpecLad:jepg #953 from jet47:fix-bug-3069 #955 from SpecLad:symlink #957 from pengx17:2.4_fix_corner_detector #959 from SpecLad:qt4-build #960 from SpecLad:extra-modules Conflicts: modules/core/include/opencv2/core/core.hpp modules/gpu/CMakeLists.txt modules/gpu/include/opencv2/gpu/device/vec_math.hpp modules/gpu/perf/perf_video.cpp modules/gpuimgproc/src/cuda/hough.cu modules/ocl/include/opencv2/ocl/ocl.hpp modules/ocl/src/pyrlk.cpp samples/gpu/driver_api_multi.cpp samples/gpu/driver_api_stereo_multi.cpp samples/ocl/surf_matcher.cpp
284 lines
8.3 KiB
C++
284 lines
8.3 KiB
C++
#include <iostream>
|
|
#include <vector>
|
|
#include <iomanip>
|
|
|
|
#include "opencv2/core/utility.hpp"
|
|
#include "opencv2/highgui/highgui.hpp"
|
|
#include "opencv2/ocl/ocl.hpp"
|
|
#include "opencv2/video/video.hpp"
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
using namespace cv::ocl;
|
|
|
|
typedef unsigned char uchar;
|
|
#define LOOP_NUM 10
|
|
int64 work_begin = 0;
|
|
int64 work_end = 0;
|
|
|
|
static void workBegin()
|
|
{
|
|
work_begin = getTickCount();
|
|
}
|
|
static void workEnd()
|
|
{
|
|
work_end += (getTickCount() - work_begin);
|
|
}
|
|
static double getTime(){
|
|
return work_end * 1000. / getTickFrequency();
|
|
}
|
|
|
|
static void download(const oclMat& d_mat, vector<Point2f>& vec)
|
|
{
|
|
vec.clear();
|
|
vec.resize(d_mat.cols);
|
|
Mat mat(1, d_mat.cols, CV_32FC2, (void*)&vec[0]);
|
|
d_mat.download(mat);
|
|
}
|
|
|
|
static void download(const oclMat& d_mat, vector<uchar>& vec)
|
|
{
|
|
vec.clear();
|
|
vec.resize(d_mat.cols);
|
|
Mat mat(1, d_mat.cols, CV_8UC1, (void*)&vec[0]);
|
|
d_mat.download(mat);
|
|
}
|
|
|
|
static void drawArrows(Mat& frame, const vector<Point2f>& prevPts, const vector<Point2f>& nextPts, const vector<uchar>& status, Scalar line_color = Scalar(0, 0, 255))
|
|
{
|
|
for (size_t i = 0; i < prevPts.size(); ++i)
|
|
{
|
|
if (status[i])
|
|
{
|
|
int line_thickness = 1;
|
|
|
|
Point p = prevPts[i];
|
|
Point q = nextPts[i];
|
|
|
|
double angle = atan2((double) p.y - q.y, (double) p.x - q.x);
|
|
|
|
double hypotenuse = sqrt( (double)(p.y - q.y)*(p.y - q.y) + (double)(p.x - q.x)*(p.x - q.x) );
|
|
|
|
if (hypotenuse < 1.0)
|
|
continue;
|
|
|
|
// Here we lengthen the arrow by a factor of three.
|
|
q.x = (int) (p.x - 3 * hypotenuse * cos(angle));
|
|
q.y = (int) (p.y - 3 * hypotenuse * sin(angle));
|
|
|
|
// Now we draw the main line of the arrow.
|
|
line(frame, p, q, line_color, line_thickness);
|
|
|
|
// Now draw the tips of the arrow. I do some scaling so that the
|
|
// tips look proportional to the main line of the arrow.
|
|
|
|
p.x = (int) (q.x + 9 * cos(angle + CV_PI / 4));
|
|
p.y = (int) (q.y + 9 * sin(angle + CV_PI / 4));
|
|
line(frame, p, q, line_color, line_thickness);
|
|
|
|
p.x = (int) (q.x + 9 * cos(angle - CV_PI / 4));
|
|
p.y = (int) (q.y + 9 * sin(angle - CV_PI / 4));
|
|
line(frame, p, q, line_color, line_thickness);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
int main(int argc, const char* argv[])
|
|
{
|
|
static std::vector<Info> ocl_info;
|
|
ocl::getDevice(ocl_info);
|
|
//if you want to use undefault device, set it here
|
|
setDevice(ocl_info[0]);
|
|
|
|
//set this to save kernel compile time from second time you run
|
|
ocl::setBinpath("./");
|
|
const char* keys =
|
|
"{ help h | false | print help message }"
|
|
"{ left l | | specify left image }"
|
|
"{ right r | | specify right image }"
|
|
"{ camera c | 0 | enable camera capturing }"
|
|
"{ use_cpu s | false | use cpu or gpu to process the image }"
|
|
"{ video v | | use video as input }"
|
|
"{ points | 1000 | specify points count [GoodFeatureToTrack] }"
|
|
"{ min_dist | 0 | specify minimal distance between points [GoodFeatureToTrack] }";
|
|
|
|
CommandLineParser cmd(argc, argv, keys);
|
|
|
|
if (cmd.has("help"))
|
|
{
|
|
cmd.printMessage();
|
|
return 0;
|
|
}
|
|
|
|
bool defaultPicturesFail = false;
|
|
string fname0 = cmd.get<string>("left");
|
|
string fname1 = cmd.get<string>("right");
|
|
string vdofile = cmd.get<string>("video");
|
|
int points = cmd.get<int>("points");
|
|
double minDist = cmd.get<double>("min_dist");
|
|
bool useCPU = cmd.has("s");
|
|
bool useCamera = cmd.has("c");
|
|
int inputName = cmd.get<int>("c");
|
|
|
|
oclMat d_nextPts, d_status;
|
|
GoodFeaturesToTrackDetector_OCL d_features(points);
|
|
Mat frame0 = imread(fname0, cv::IMREAD_GRAYSCALE);
|
|
Mat frame1 = imread(fname1, cv::IMREAD_GRAYSCALE);
|
|
PyrLKOpticalFlow d_pyrLK;
|
|
vector<cv::Point2f> pts(points);
|
|
vector<cv::Point2f> nextPts(points);
|
|
vector<unsigned char> status(points);
|
|
vector<float> err;
|
|
|
|
if (frame0.empty() || frame1.empty())
|
|
{
|
|
useCamera = true;
|
|
defaultPicturesFail = true;
|
|
VideoCapture capture(inputName);
|
|
if (!capture.isOpened())
|
|
{
|
|
cout << "Can't load input images" << endl;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
cout << "Points count : " << points << endl << endl;
|
|
|
|
if (useCamera)
|
|
{
|
|
VideoCapture capture;
|
|
Mat frame, frameCopy;
|
|
Mat frame0Gray, frame1Gray;
|
|
Mat ptr0, ptr1;
|
|
|
|
if(vdofile == "")
|
|
capture.open( inputName );
|
|
else
|
|
capture.open(vdofile.c_str());
|
|
|
|
int c = inputName ;
|
|
if(!capture.isOpened())
|
|
{
|
|
if(vdofile == "")
|
|
cout << "Capture from CAM " << c << " didn't work" << endl;
|
|
else
|
|
cout << "Capture from file " << vdofile << " failed" <<endl;
|
|
if (defaultPicturesFail)
|
|
{
|
|
return -1;
|
|
}
|
|
goto nocamera;
|
|
}
|
|
|
|
cout << "In capture ..." << endl;
|
|
for(int i = 0;; i++)
|
|
{
|
|
if( !capture.read(frame) )
|
|
break;
|
|
|
|
if (i == 0)
|
|
{
|
|
frame.copyTo( frame0 );
|
|
cvtColor(frame0, frame0Gray, COLOR_BGR2GRAY);
|
|
}
|
|
else
|
|
{
|
|
if (i%2 == 1)
|
|
{
|
|
frame.copyTo(frame1);
|
|
cvtColor(frame1, frame1Gray, COLOR_BGR2GRAY);
|
|
ptr0 = frame0Gray;
|
|
ptr1 = frame1Gray;
|
|
}
|
|
else
|
|
{
|
|
frame.copyTo(frame0);
|
|
cvtColor(frame0, frame0Gray, COLOR_BGR2GRAY);
|
|
ptr0 = frame1Gray;
|
|
ptr1 = frame0Gray;
|
|
}
|
|
|
|
if (useCPU)
|
|
{
|
|
pts.clear();
|
|
goodFeaturesToTrack(ptr0, pts, points, 0.01, 0.0);
|
|
if(pts.size() == 0)
|
|
continue;
|
|
calcOpticalFlowPyrLK(ptr0, ptr1, pts, nextPts, status, err);
|
|
}
|
|
else
|
|
{
|
|
oclMat d_img(ptr0), d_prevPts;
|
|
d_features(d_img, d_prevPts);
|
|
if(!d_prevPts.rows || !d_prevPts.cols)
|
|
continue;
|
|
d_pyrLK.sparse(d_img, oclMat(ptr1), d_prevPts, d_nextPts, d_status);
|
|
d_features.downloadPoints(d_prevPts,pts);
|
|
download(d_nextPts, nextPts);
|
|
download(d_status, status);
|
|
}
|
|
if (i%2 == 1)
|
|
frame1.copyTo(frameCopy);
|
|
else
|
|
frame0.copyTo(frameCopy);
|
|
drawArrows(frameCopy, pts, nextPts, status, Scalar(255, 0, 0));
|
|
imshow("PyrLK [Sparse]", frameCopy);
|
|
}
|
|
|
|
if( waitKey( 10 ) >= 0 )
|
|
goto _cleanup_;
|
|
}
|
|
|
|
waitKey(0);
|
|
|
|
_cleanup_:
|
|
capture.release();
|
|
}
|
|
else
|
|
{
|
|
nocamera:
|
|
for(int i = 0; i <= LOOP_NUM;i ++)
|
|
{
|
|
cout << "loop" << i << endl;
|
|
if (i > 0) workBegin();
|
|
|
|
if (useCPU)
|
|
{
|
|
goodFeaturesToTrack(frame0, pts, points, 0.01, minDist);
|
|
calcOpticalFlowPyrLK(frame0, frame1, pts, nextPts, status, err);
|
|
}
|
|
else
|
|
{
|
|
oclMat d_img(frame0), d_prevPts;
|
|
d_features(d_img, d_prevPts);
|
|
d_pyrLK.sparse(d_img, oclMat(frame1), d_prevPts, d_nextPts, d_status);
|
|
d_features.downloadPoints(d_prevPts, pts);
|
|
download(d_nextPts, nextPts);
|
|
download(d_status, status);
|
|
}
|
|
|
|
if (i > 0 && i <= LOOP_NUM)
|
|
workEnd();
|
|
|
|
if (i == LOOP_NUM)
|
|
{
|
|
if (useCPU)
|
|
cout << "average CPU time (noCamera) : ";
|
|
else
|
|
cout << "average GPU time (noCamera) : ";
|
|
|
|
cout << getTime() / LOOP_NUM << " ms" << endl;
|
|
|
|
drawArrows(frame0, pts, nextPts, status, Scalar(255, 0, 0));
|
|
|
|
imshow("PyrLK [Sparse]", frame0);
|
|
}
|
|
}
|
|
}
|
|
|
|
waitKey();
|
|
|
|
return 0;
|
|
}
|