generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

video writing via cmd args in hog sample added

Browse Source
This commit is contained in:
Alexey Spizhevoy 2010-12-30 07:19:32 +00:00
parent 4ad938afcc
commit 1a34fa30f4
1 changed files with 54 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
samples/gpu/hog.cpp
View File

@ -10,8 +10,6 @@
using namespace std; using namespace std;
using namespace cv; using namespace cv;
//#define WRITE_VIDEO
class Args class Args
{ {
public: public:
@ -23,15 +21,21 @@ public:
bool src_is_camera; bool src_is_camera;
int camera_id; int camera_id;
bool write_video;
string dst_video;
double dst_video_fps;
bool make_gray; bool make_gray;
bool resize_src; bool resize_src;
int resized_width, resized_height; int width, height;
double scale; double scale;
int nlevels; int nlevels;
int gr_threshold; int gr_threshold;
double hit_threshold; double hit_threshold;
bool hit_threshold_auto;
int win_width; int win_width;
int win_stride_width, win_stride_height; int win_stride_width, win_stride_height;
@ -93,8 +97,8 @@ int main(int argc, char** argv)
<< " [--src-is-camera <true/false>] # says to interpretate src as camera\n" << " [--src-is-camera <true/false>] # says to interpretate src as camera\n"
<< " [--make-gray <true/false>] # convert image to gray one or not\n" << " [--make-gray <true/false>] # convert image to gray one or not\n"
<< " [--resize-src <true/false>] # do resize of the source image or not\n" << " [--resize-src <true/false>] # do resize of the source image or not\n"
<< " [--src-width <int>] # resized image width\n" << " [--width <int>] # resized image width\n"
<< " [--src-height <int>] # resized image height\n" << " [--height <int>] # resized image height\n"
<< " [--hit-threshold <double>] # classifying plane distance threshold (0.0 usually)\n" << " [--hit-threshold <double>] # classifying plane distance threshold (0.0 usually)\n"
<< " [--scale <double>] # HOG window scale factor\n" << " [--scale <double>] # HOG window scale factor\n"
<< " [--nlevels <int>] # max number of HOG window scales\n" << " [--nlevels <int>] # max number of HOG window scales\n"
@ -102,15 +106,18 @@ int main(int argc, char** argv)
<< " [--win-stride-width <int>] # distance by OX axis between neighbour wins\n" << " [--win-stride-width <int>] # distance by OX axis between neighbour wins\n"
<< " [--win-stride-height <int>] # distance by OY axis between neighbour wins\n" << " [--win-stride-height <int>] # distance by OY axis between neighbour wins\n"
<< " [--gr-threshold <int>] # merging similar rects constant\n" << " [--gr-threshold <int>] # merging similar rects constant\n"
<< " [--gamma-correct <int>] # do gamma correction or not\n"; << " [--gamma-correct <int>] # do gamma correction or not\n"
<< " [--write-video <bool>] # write video or not\n"
<< " [--dst-video <path>] # output video path\n"
<< " [--dst-video-fps <double>] # output video fps\n";
return 1; return 1;
} }
App app(Args::read(argc, argv)); App app(Args::read(argc, argv));
app.run(); app.run();
} }
catch (const Exception& e) { return cout << "Error: " << e.what() << endl, 1; } catch (const Exception& e) { return cout << "error: " << e.what() << endl, 1; }
catch (const exception& e) { return cout << "Error: " << e.what() << endl, 1; } catch (const exception& e) { return cout << "error: " << e.what() << endl, 1; }
catch(...) { return cout << "Unknown exception" << endl, 1; } catch(...) { return cout << "unknown exception" << endl, 1; }
return 0; return 0;
} }
@ -121,16 +128,20 @@ Args::Args()
src_is_camera = false; src_is_camera = false;
camera_id = 0; camera_id = 0;
write_video = false;
dst_video_fps = 24.;
make_gray = false; make_gray = false;
resize_src = false; resize_src = false;
resized_width = 640; width = 640;
resized_height = 480; height = 480;
scale = 1.05; scale = 1.05;
nlevels = 13; nlevels = 13;
gr_threshold = 8; gr_threshold = 8;
hit_threshold = 1.4; hit_threshold = 1.4;
hit_threshold_auto = true;
win_width = 48; win_width = 48;
win_stride_width = 8; win_stride_width = 8;
@ -153,16 +164,23 @@ Args Args::read(int argc, char** argv)
else if (key == "--camera-id") args.camera_id = atoi(val.c_str()); else if (key == "--camera-id") args.camera_id = atoi(val.c_str());
else if (key == "--make-gray") args.make_gray = (val == "true"); else if (key == "--make-gray") args.make_gray = (val == "true");
else if (key == "--resize-src") args.resize_src = (val == "true"); else if (key == "--resize-src") args.resize_src = (val == "true");
else if (key == "--src-width") args.resized_width = atoi(val.c_str()); else if (key == "--width") args.width = atoi(val.c_str());
else if (key == "--src-height") args.resized_height = atoi(val.c_str()); else if (key == "--height") args.height = atoi(val.c_str());
else if (key == "--hit-threshold") args.hit_threshold = atof(val.c_str()); else if (key == "--hit-threshold")
{
args.hit_threshold = atof(val.c_str());
args.hit_threshold_auto = false;
}
else if (key == "--scale") args.scale = atof(val.c_str()); else if (key == "--scale") args.scale = atof(val.c_str());
else if (key == "--nlevels") args.nlevels = atoi(val.c_str()); else if (key == "--nlevels") args.nlevels = atoi(val.c_str());
else if (key == "--win-width") args.win_width = atoi(val.c_str()); else if (key == "--win-width") args.win_width = atoi(val.c_str());
else if (key == "--win-stride-width") args.win_stride_width = atoi(val.c_str()); else if (key == "--win-stride-width") args.win_stride_width = atoi(val.c_str());
else if (key == "--win-stride-height") args.win_stride_height = atoi(val.c_str()); else if (key == "--win-stride-height") args.win_stride_height = atoi(val.c_str());
else if (key == "--gr-threshold") args.gr_threshold = atoi(val.c_str()); else if (key == "--gr-threshold") args.gr_threshold = atoi(val.c_str());
else if (key == "--gamma-correct") args.gamma_corr = atoi(val.c_str()) != 0; else if (key == "--gamma-correct") args.gamma_corr = (val == "true");
else if (key == "--write-video") args.write_video = (val == "true");
else if (key == "--dst-video") args.dst_video = val;
else if (key == "--dst-video-fps") args.dst_video_fps= atof(val.c_str());
else throw runtime_error((string("unknown key: ") + key)); else throw runtime_error((string("unknown key: ") + key));
} }
return args; return args;
@ -187,7 +205,11 @@ App::App(const Args& s)
scale = args.scale; scale = args.scale;
gr_threshold = args.gr_threshold; gr_threshold = args.gr_threshold;
nlevels = args.nlevels; nlevels = args.nlevels;
if (args.hit_threshold_auto)
args.hit_threshold = args.win_width == 48 ? 1.4 : 0.;
hit_threshold = args.hit_threshold; hit_threshold = args.hit_threshold;
gamma_corr = args.gamma_corr; gamma_corr = args.gamma_corr;
if (args.win_width != 64 && args.win_width != 48) if (args.win_width != 64 && args.win_width != 48)
@ -195,7 +217,7 @@ App::App(const Args& s)
cout << "Scale: " << scale << endl; cout << "Scale: " << scale << endl;
if (args.resize_src) if (args.resize_src)
cout << "Source size: (" << args.resized_width << ", " << args.resized_height << ")\n"; cout << "Resized source: (" << args.width << ", " << args.height << ")\n";
cout << "Group threshold: " << gr_threshold << endl; cout << "Group threshold: " << gr_threshold << endl;
cout << "Levels number: " << nlevels << endl; cout << "Levels number: " << nlevels << endl;
cout << "Win width: " << args.win_width << endl; cout << "Win width: " << args.win_width << endl;
@ -209,10 +231,12 @@ App::App(const Args& s)
void App::run() void App::run()
{ {
running = true; running = true;
cv::VideoWriter video_writer;
Size win_size(args.win_width, args.win_width * 2); //(64, 128) or (48, 96) Size win_size(args.win_width, args.win_width * 2); //(64, 128) or (48, 96)
Size win_stride(args.win_stride_width, args.win_stride_height); Size win_stride(args.win_stride_width, args.win_stride_height);
// Create HOG descriptors and detectors here
vector<float> detector; vector<float> detector;
if (win_size == Size(64, 128)) if (win_size == Size(64, 128))
detector = cv::gpu::HOGDescriptor::getPeopleDetector_64x128(); detector = cv::gpu::HOGDescriptor::getPeopleDetector_64x128();
@ -227,13 +251,6 @@ void App::run()
gpu_hog.setSVMDetector(detector); gpu_hog.setSVMDetector(detector);
cpu_hog.setSVMDetector(detector); cpu_hog.setSVMDetector(detector);
#ifdef WRITE_VIDEO
cv::VideoWriter video_writer;
video_writer.open("output.avi", CV_FOURCC('x','v','i','d'), 24., cv::Size(640, 480), true);
if (!video_writer.isOpened())
throw std::runtime_error("can't create video writer");
#endif
while (running) while (running)
{ {
VideoCapture vc; VideoCapture vc;
@ -274,7 +291,7 @@ void App::run()
else img_aux = frame; else img_aux = frame;
// Resize image // Resize image
if (args.resize_src) resize(img_aux, img, Size(args.resized_width, args.resized_height)); if (args.resize_src) resize(img_aux, img, Size(args.width, args.height));
else img = img_aux; else img = img_aux;
img_to_show = img; img_to_show = img;
@ -315,10 +332,18 @@ void App::run()
workEnd(); workEnd();
#ifdef WRITE_VIDEO if (args.write_video)
cvtColor(img_to_show, img, CV_BGRA2BGR); {
video_writer << img; if (!video_writer.isOpened())
#endif {
video_writer.open(args.dst_video, CV_FOURCC('x','v','i','d'), args.dst_video_fps,
img_to_show.size(), true);
if (!video_writer.isOpened())
throw std::runtime_error("can't create video writer");
}
cvtColor(img_to_show, img, CV_BGRA2BGR);
video_writer << img;
}
} }
} }
} }
@ -418,3 +443,4 @@ inline string App::workFps() const
ss << work_fps; ss << work_fps;
return ss.str(); return ss.str();
} }