mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-27 23:11:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add the possibility to use a video stream with the peopledetect.cpp sample.

Browse Source 
Fixed video input argument.
This commit is contained in:
catree 2016-09-04 04:17:15 +02:00
parent d102ea96c0
commit aaa335b617
1 changed files with 90 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
samples/cpp/peopledetect.cpp
View File

@ -1,27 +1,23 @@
#include <iostream>
#include <stdexcept>
#include <opencv2/objdetect.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <iostream>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/video.hpp>
#include <opencv2/videoio.hpp>
using namespace cv;
using namespace std;
static void help()
{
cout << "\nThis program demonstrates the use of the HoG descriptor using\n"
" HOGDescriptor::hog.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());\n"
"Usage:\n"
"./peopledetect --i=<image_filename> | --d=<image_directory>\n\n"
"During execution:\n\tHit q or ESC key to quit.\n"
"\tUsing OpenCV version " << CV_VERSION << "\n" << endl;
}
const char* keys =
{
"{help h||}"
"{image i| ../data/basketball2.png|input image name}"
"{directory d||images directory}"
"{ help h | | print help message }"
"{ image i | | specify input image}"
"{ camera c | | enable camera capturing }"
"{ video v | ../data/768x576.avi | use video as input }"
"{ directory d | | images directory}"
};
static void detectAndDraw(const HOGDescriptor &hog, Mat &img)
@ -69,7 +65,12 @@ int main(int argc, char** argv)
if (parser.has("help"))
{
help();
cout << "\nThis program demonstrates the use of the HoG descriptor using\n"
" HOGDescriptor::hog.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());\n";
parser.printMessage();
cout << "During execution:\n\tHit q or ESC key to quit.\n"
"\tUsing OpenCV version " << CV_VERSION << "\n"
"Note: camera device number must be different from -1.\n" << endl;
return 0;
}
@ -78,6 +79,8 @@ int main(int argc, char** argv)
namedWindow("people detector", 1);
string pattern_glob = "";
string video_filename = "../data/768x576.avi";
int camera_id = -1;
if (parser.has("directory"))
{
pattern_glob = parser.get<string>("directory");
@ -86,26 +89,85 @@ int main(int argc, char** argv)
{
pattern_glob = parser.get<string>("image");
}
else if (parser.has("camera"))
{
camera_id = parser.get<int>("camera");
}
else if (parser.has("video"))
{
video_filename = parser.get<string>("video");
}
if (!pattern_glob.empty() || camera_id != -1 || !video_filename.empty())
{
//Read from input image files
vector<String> filenames;
//Read from video file
VideoCapture vc;
Mat frame;
if (!pattern_glob.empty())
{
vector<String> filenames;
String folder(pattern_glob);
glob(folder, filenames);
for (vector<String>::const_iterator it = filenames.begin(); it != filenames.end(); ++it)
}
else if (camera_id != -1)
{
cout << "\nRead: " << *it << endl;
vc.open(camera_id);
if (!vc.isOpened())
{
stringstream msg;
msg << "can't open camera: " << camera_id;
throw runtime_error(msg.str());
}
}
else
{
vc.open(video_filename.c_str());
if (!vc.isOpened())
throw runtime_error(string("can't open video file: " + video_filename));
}
vector<String>::const_iterator it_image = filenames.begin();
for (;;)
{
if (!pattern_glob.empty())
{
bool read_image_ok = false;
for (; it_image != filenames.end(); ++it_image)
{
cout << "\nRead: " << *it_image << endl;
// Read current image
Mat current_image = imread(*it);
frame = imread(*it_image);
if (current_image.empty())
continue;
if (!frame.empty())
{
++it_image;
read_image_ok = true;
break;
}
}
detectAndDraw(hog, current_image);
//No more valid images
if (!read_image_ok)
{
//Release the image in order to exit the while loop
frame.release();
}
}
else
{
vc >> frame;
}
imshow("people detector", current_image);
int c = waitKey(0) & 255;
if (frame.empty())
break;
detectAndDraw(hog, frame);
imshow("people detector", frame);
int c = waitKey( vc.isOpened() ? 30 : 0 ) & 255;
if ( c == 'q' || c == 'Q' || c == 27)
break;
}