a good template for people to start quickly with image lists and video processingpull/13383/head
parent
88e6ae4c36
commit
22d9f3558f
1 changed files with 106 additions and 0 deletions
@ -0,0 +1,106 @@ |
||||
/*
|
||||
* cvstarter.cpp |
||||
* |
||||
* Created on: Nov 23, 2010 |
||||
* Author: Ethan Rublee |
||||
* |
||||
* A starter sample for using opencv, load up a list or get a video stream |
||||
* easy as CV_PI right? |
||||
*/ |
||||
#include <opencv2/opencv.hpp> |
||||
#include <vector> |
||||
|
||||
using namespace cv; |
||||
using namespace std; |
||||
|
||||
//hide the local functions in an anon namespace
|
||||
namespace |
||||
{ |
||||
void help(char** av) |
||||
{ |
||||
cout << "usage:" << av[0] << " image_list.yaml\n" |
||||
<< "\tThis is a starter sample, to get you up and going in a copy pasta fashion\n" |
||||
<< "\tTry running imagelist_creator to generate a list of images\n" |
||||
<< "\tOr try running this with an integer arg like:\n\n" << av[0] << " 0\n" |
||||
<< "\tThis will open up the video device." << endl; |
||||
} |
||||
|
||||
bool readStringList(const string& filename, vector<string>& l) |
||||
{ |
||||
l.resize(0); |
||||
FileStorage fs(filename, FileStorage::READ); |
||||
if (!fs.isOpened()) |
||||
return false; |
||||
FileNode n = fs.getFirstTopLevelNode(); |
||||
if (n.type() != FileNode::SEQ) |
||||
return false; |
||||
FileNodeIterator it = n.begin(), it_end = n.end(); |
||||
for (; it != it_end; ++it) |
||||
l.push_back((string)*it); |
||||
return true; |
||||
} |
||||
|
||||
int videoProcess(VideoCapture& capture) |
||||
{ |
||||
|
||||
namedWindow("video",CV_WINDOW_KEEPRATIO); //resizable window;
|
||||
Mat frame; |
||||
for (;;) |
||||
{ |
||||
capture >> frame; |
||||
if (frame.empty()) |
||||
continue; |
||||
imshow("video",frame); |
||||
char key = waitKey(5); //delay N millis, usually long enough to display and capture input
|
||||
switch (key) |
||||
{ |
||||
case 'q': |
||||
case 'Q': |
||||
case 27: //escape key
|
||||
return 0; |
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
int imageListProcess(vector<string> images) |
||||
{ |
||||
namedWindow("image",CV_WINDOW_KEEPRATIO); //resizable window;
|
||||
for (size_t i = 0; i < images.size(); i++) |
||||
{ |
||||
Mat image = imread(images[i], CV_LOAD_IMAGE_GRAYSCALE); // do grayscale processing?
|
||||
imshow("image",image); |
||||
cout << "Press a key to see the next image in the list." << endl; |
||||
waitKey(); // wait indefinitely
|
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
} |
||||
|
||||
int main(int ac, char** av) |
||||
{ |
||||
|
||||
if (ac != 2) |
||||
{ |
||||
help(av); |
||||
return 1; |
||||
} |
||||
std::string arg = av[1]; |
||||
vector<string> imagelist; |
||||
|
||||
if (!readStringList(arg,imagelist)) |
||||
{ |
||||
int ad = atoi(arg.c_str()); |
||||
VideoCapture capture(ad); |
||||
return videoProcess(capture); |
||||
} |
||||
else |
||||
{ |
||||
return imageListProcess(imagelist); |
||||
} |
||||
|
||||
return 0; |
||||
} |
Loading…
Reference in new issue