mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-07 17:44:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

opencv_visualization: check cmdline args

Browse Source
This commit is contained in:
berak 2016-07-12 07:38:52 +02:00
parent 630406e006
commit 20b9ff4ff6
1 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
apps/visualisation/opencv_visualisation.cpp
View File

@ -47,7 +47,7 @@ Software for visualising cascade classifier models trained by OpenCV and to get
understanding of the used features. understanding of the used features.
USAGE: USAGE:
./visualise_models -model <model.xml> -image <ref.png> -data <output folder> ./opencv_visualisation --model=<model.xml> --image=<ref.png> --data=<video output folder>
LIMITS LIMITS
- Use an absolute path for the output folder to ensure the tool works - Use an absolute path for the output folder to ensure the tool works
@ -81,20 +81,23 @@ struct rect_data{
int main( int argc, const char** argv ) int main( int argc, const char** argv )
{ {
CommandLineParser parser(argc, argv,
"{ help h usage ? | | show this message }"
"{ image i | | (required) path to reference image }"
"{ model m | | (required) path to cascade xml file }"
"{ data d | | (optional) path to video output folder }"
);
// Read in the input arguments // Read in the input arguments
string model = ""; if (parser.has("help")){
string output_folder = ""; parser.printMessage();
string image_ref = ""; return 0;
for(int i = 1; i < argc; ++i ) }
{ string model(parser.get<string>("model"));
if( !strcmp( argv[i], "-model" ) ) string output_folder(parser.get<string>("data"));
{ string image_ref = (parser.get<string>("image"));
model = argv[++i]; if (model.empty() || image_ref.empty()){
}else if( !strcmp( argv[i], "-image" ) ){ parser.printMessage();
image_ref = argv[++i]; return -1;
}else if( !strcmp( argv[i], "-data" ) ){
output_folder = argv[++i];
}
} }
// Value for timing // Value for timing
@ -106,8 +109,11 @@ int main( int argc, const char** argv )
// Open the XML model // Open the XML model
FileStorage fs; FileStorage fs;
fs.open(model, FileStorage::READ); bool model_ok = fs.open(model, FileStorage::READ);
if (!model_ok){
cerr << "the cascade file '" << model << "' could not be loaded." << endl;
return -1;
}
// Get a the required information // Get a the required information
// First decide which feature type we are using // First decide which feature type we are using
FileNode cascade = fs["cascade"]; FileNode cascade = fs["cascade"];
@ -129,6 +135,10 @@ int main( int argc, const char** argv )
int resize_factor = 10; int resize_factor = 10;
int resize_storage_factor = 10; int resize_storage_factor = 10;
Mat reference_image = imread(image_ref, IMREAD_GRAYSCALE ); Mat reference_image = imread(image_ref, IMREAD_GRAYSCALE );
if (reference_image.empty()){
cerr << "the reference image '" << image_ref << "'' could not be loaded." << endl;
return -1;
}
Mat visualization; Mat visualization;
resize(reference_image, visualization, Size(reference_image.cols * resize_factor, reference_image.rows * resize_factor)); resize(reference_image, visualization, Size(reference_image.cols * resize_factor, reference_image.rows * resize_factor));