MyException - 我的异常网
当前位置:我的异常网» 编程 » opencv入门札记之四 读写视频 捕捉移动点

opencv入门札记之四 读写视频 捕捉移动点

www.MyException.Cn  网友分享于:2015-02-11  浏览:0次
opencv入门笔记之四 读写视频 捕捉移动点

 

 捕捉移动点时,  我们可以使用一种叫做 光学流动(optical flow) 的算法,  在关键点的周围,  所有点的流动导数都是相同的,  从而我门可以判断出关键点( feature points)

 下面是代码:

classCameraCaliberator

{

vector<vector<Point3f>> objectPoints;

// the real points

vector<vector<Point2f>> imagePoints;

// points found inimage

Mat cameraMatrix;

Mat distCoeffs;

int flag;

Mat map1, map2;

bool mustInitUndistort;

 

public:

CameraCaliberator() :flag(0), mustInitUndistort(true)

{

};

 

int addChessboardPoints(

const vector<string>& filelist, Size &boardSize){

vector<Point2f> imageCorners;

vector<Point3f> objectCorners;

for (int i = 0; i < boardSize.height;i++)

{

for (int j = 0; j < boardSize.width;j++)

{

objectCorners.push_back(Point3f(i, j,0.0f));        

}

}

Mat image;

int successes = 0;

for (int i = 0; i < filelist.size();i++)

{

image = imread(filelist[i], 0);

 

bool found = findChessboardCorners(image, boardSize, imageCorners);

cornerSubPix(image, imageCorners, Size(5, 5), Size(-1, -1),

TermCriteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 30, 0.1));

//used to define when to stop

if (imageCorners.size() == boardSize.area())

{

addPoints(imageCorners, objectCorners);

successes++;

}

}

return successes;

}

double calibrate(Size &imageSize){

mustInitUndistort = true;

vector<Mat> rvecs, tvecs;

return        calibrateCamera(objectPoints,

imagePoints,

imageSize,

cameraMatrix,

distCoeffs,

rvecs,

tvecs,

flag);

}

Mat remap(const        Mat&image){

Mat undistorted;

if (mustInitUndistort){

initUndistortRectifyMap(cameraMatrix,

distCoeffs,

Mat(), Mat(),

image.size(),

CV_32FC1,

map1, map2);

 

}

mustInitUndistort = false;

cv::remap(image, undistorted, map1, map2, INTER_LINEAR);

return undistorted;

};

//        // Set the calibration options

//        // 8radialCoeffEnabled shouldbe true if 8 radial coefficients are required (5 is default)

//        // tangentialParamEnabledshould be true if tangeantial distortion is present

//        voidCameraCaliberator::setCalibrationFlag(bool radial8CoeffEnabled, booltangentialParamEnabled) {

//

//                //Set the flag used in cv::calibrateCamera()

//                flag= 0;

//                if(!tangentialParamEnabled) flag += CV_CALIB_ZERO_TANGENT_DIST;

//                if(radial8CoeffEnabled) flag += CV_CALIB_RATIONAL_MODEL;

//        }

 

 

private:

void addPoints(vector<Point2f>& imageCorners,vector<Point3f>& objectCorners){

imagePoints.push_back(imageCorners);

objectPoints.push_back(objectCorners);

}

};

 

 

 

void m(){

vector<string> filelist;

Size boardSize(6, 4);

for (int i = 1; i <= 20; i++) {

 

stringstream str;

str << "chessboards\\chessboard" <<std::setw(2) << std::setfill('0') << i << ".jpg";

cout << str.str() << endl;

 

filelist.push_back(str.str());

//                image= cv::imread(str.str(), 0);

//                cv::imshow("Image",image);

//

//                cv::waitKey(100);

}

Mat image=imread(filelist[0]);

CameraCaliberator cameraCaliberator;

 

cameraCaliberator.addChessboardPoints(filelist,boardSize);

cameraCaliberator.calibrate(image.size());

Matuimage=        cameraCaliberator.remap(image);

imshow(" ", uimage);

}

 

 

 

// Openthe video file

cv::VideoCapturecapture("bike.avi");

// checkif video successfully opened

if(!capture.isOpened())

return 1;

// Getthe frame rate

doublerate = capture.get(CV_CAP_PROP_FPS);

boolstop(false);

cv::Matframe; // current video frame

cv::namedWindow("ExtractedFrame");

// Delaybetween each frame in ms

//corresponds to video frame rate

int delay= 1000 / rate;

// forall frames in video

while(!stop) {

// read next frame if any

if (!capture.read(frame))

break;

cv::imshow("Extracted Frame", frame);

// introduce a delay

// or press key to stop

if (cv::waitKey(delay) >= 0)

stop = true;

}

// Closethe video file.

// Notrequired since called by destructor

capture.release();

 

 

 

class FeatureTracker : public FrameProcessor {

 

cv::Matgray;                        //current gray-level image

cv::Matgray_prev;                //previous gray-level image

std::vector<cv::Point2f> points[2]; // tracked features from0->1

std::vector<cv::Point2f> initial;   // initial position of tracked points

std::vector<cv::Point2f> features;  // detected features

int max_count;          // maximum number of features to detect

double qlevel;    // qualitylevel for feature detection

double minDist;   // minimumdistance between two feature points

std::vector<uchar> status; // status of tracked features

    std::vector<float> err;    // error in tracking

 

  public:

 

FeatureTracker() : max_count(500), qlevel(0.01), minDist(10.) {}

 

// processing method

void process(cv:: Mat &frame, cv:: Mat &output) {

 

// convert to gray-level image

cv::cvtColor(frame, gray, CV_BGR2GRAY);

frame.copyTo(output);

 

// 1. if new feature points must be added

if(addNewPoints())

{

// detect feature points

detectFeaturePoints();

// add the detected features to the currently tracked features

points[0].insert(points[0].end(),features.begin(),features.end());

initial.insert(initial.end(),features.begin(),features.end());

}

 

// for first image of the sequence

if(gray_prev.empty())

           gray.copyTo(gray_prev);

           

// 2. track features

cv::calcOpticalFlowPyrLK(gray_prev, gray, // 2 consecutive images

points[0], // input point position in first image

points[1], // output point postion in the second image

status,    // tracking success

err);      // tracking error

          

// 2. loop over the tracked points to reject the undesirables

int k=0;

for( int i= 0; i < points[1].size(); i++ ) {

 

// do we keep this point?

if (acceptTrackedPoint(i)) {

 

// keep this point in vector

initial[k]= initial[i];

points[1][k++] = points[1][i];

}

}

 

// eliminate unsuccesful points

        points[1].resize(k);

initial.resize(k);

 

// 3. handle the accepted tracked points

handleTrackedPoints(frame, output);

 

// 4. current points and image become previous ones

std::swap(points[1], points[0]);

        cv::swap(gray_prev, gray);

}

 

// feature point detection

void detectFeaturePoints() {

 

// detect the features

cv::goodFeaturesToTrack(gray, // the image

features,   // the outputdetected features

max_count,  // the maximumnumber of features

qlevel,     // quality level

minDist);   // min distancebetween two features

}

 

// determine if new points should be added

bool addNewPoints() {

 

// if too few points

return points[0].size()<=10;

}

 

// determine which tracked point should be accepted

bool acceptTrackedPoint(int i) {

 

return status[i] &&

// if point has moved

(abs(points[0][i].x-points[1][i].x)+

(abs(points[0][i].y-points[1][i].y))>2);

}

 

// handle the currently tracked points

void handleTrackedPoints(cv:: Mat &frame, cv:: Mat &output){

 

// for all tracked points

for(int i= 0; i < points[1].size(); i++ ) {

 

// draw line and circle

    cv::line(output,initial[i], points[1][i], cv::Scalar(255,255,255));

cv::circle(output, points[1][i], 3, cv::Scalar(255,255,255),-1);

}

}

};

 

#endif

 

文章评论

Java 与 .NET 的平台发展之争
Java 与 .NET 的平台发展之争
漫画:程序员的工作
漫画:程序员的工作
程序员应该关注的一些事儿
程序员应该关注的一些事儿
中美印日四国程序员比较
中美印日四国程序员比较
那些性感的让人尖叫的程序员
那些性感的让人尖叫的程序员
程序员周末都喜欢做什么?
程序员周末都喜欢做什么?
做程序猿的老婆应该注意的一些事情
做程序猿的老婆应该注意的一些事情
鲜为人知的编程真相
鲜为人知的编程真相
我的丈夫是个程序员
我的丈夫是个程序员
当下全球最炙手可热的八位少年创业者
当下全球最炙手可热的八位少年创业者
Web开发人员为什么越来越懒了?
Web开发人员为什么越来越懒了?
为什么程序员都是夜猫子
为什么程序员都是夜猫子
我跳槽是因为他们的显示器更大
我跳槽是因为他们的显示器更大
程序猿的崛起——Growth Hacker
程序猿的崛起——Growth Hacker
总结2014中国互联网十大段子
总结2014中国互联网十大段子
程序员都该阅读的书
程序员都该阅读的书
Web开发者需具备的8个好习惯
Web开发者需具备的8个好习惯
我是如何打败拖延症的
我是如何打败拖延症的
代码女神横空出世
代码女神横空出世
亲爱的项目经理,我恨你
亲爱的项目经理,我恨你
Java程序员必看电影
Java程序员必看电影
聊聊HTTPS和SSL/TLS协议
聊聊HTTPS和SSL/TLS协议
2013年美国开发者薪资调查报告
2013年美国开发者薪资调查报告
编程语言是女人
编程语言是女人
每天工作4小时的程序员
每天工作4小时的程序员
程序员眼里IE浏览器是什么样的
程序员眼里IE浏览器是什么样的
程序员必看的十大电影
程序员必看的十大电影
团队中“技术大拿”并非越多越好
团队中“技术大拿”并非越多越好
旅行,写作,编程
旅行,写作,编程
10个帮程序员减压放松的网站
10个帮程序员减压放松的网站
初级 vs 高级开发者 哪个性价比更高?
初级 vs 高级开发者 哪个性价比更高?
十大编程算法助程序员走上高手之路
十大编程算法助程序员走上高手之路
程序员和编码员之间的区别
程序员和编码员之间的区别
什么才是优秀的用户界面设计
什么才是优秀的用户界面设计
写给自己也写给你 自己到底该何去何从
写给自己也写给你 自己到底该何去何从
科技史上最臭名昭著的13大罪犯
科技史上最臭名昭著的13大罪犯
“懒”出效率是程序员的美德
“懒”出效率是程序员的美德
老美怎么看待阿里赴美上市
老美怎么看待阿里赴美上市
程序员的鄙视链
程序员的鄙视链
5款最佳正则表达式编辑调试器
5款最佳正则表达式编辑调试器
如何区分一个程序员是“老手“还是“新手“?
如何区分一个程序员是“老手“还是“新手“?
“肮脏的”IT工作排行榜
“肮脏的”IT工作排行榜
 程序员的样子
程序员的样子
要嫁就嫁程序猿—钱多话少死的早
要嫁就嫁程序猿—钱多话少死的早
看13位CEO、创始人和高管如何提高工作效率
看13位CEO、创始人和高管如何提高工作效率
2013年中国软件开发者薪资调查报告
2013年中国软件开发者薪资调查报告
老程序员的下场
老程序员的下场
60个开发者不容错过的免费资源库
60个开发者不容错过的免费资源库
如何成为一名黑客
如何成为一名黑客
软件开发程序错误异常ExceptionCopyright © 2009-2015 MyException 版权所有