diff --git a/modules/datasets/samples/pd_inria.cpp b/modules/datasets/samples/pd_inria.cpp index fb711176a..da86e5ca8 100644 --- a/modules/datasets/samples/pd_inria.cpp +++ b/modules/datasets/samples/pd_inria.cpp @@ -1,50 +1,7 @@ -/*M/////////////////////////////////////////////////////////////////////////////////////// -// -// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. -// -// By downloading, copying, installing or using the software you agree to this license. -// If you do not agree to this license, do not download, install, -// copy or use the software. -// -// -// License Agreement -// For Open Source Computer Vision Library -// -// Copyright (C) 2015, Itseez Inc, all rights reserved. -// Third party copyrights are property of their respective owners. -// -// Redistribution and use in source and binary forms, with or without modification, -// are permitted provided that the following conditions are met: -// -// * Redistribution's of source code must retain the above copyright notice, -// this list of conditions and the following disclaimer. -// -// * Redistribution's in binary form must reproduce the above copyright notice, -// this list of conditions and the following disclaimer in the documentation -// and/or other materials provided with the distribution. -// -// * The name of the copyright holders may not be used to endorse or promote products -// derived from this software without specific prior written permission. -// -// This software is provided by the copyright holders and contributors "as is" and -// any express or implied warranties, including, but not limited to, the implied -// warranties of merchantability and fitness for a particular purpose are disclaimed. -// In no event shall the Itseez Inc or contributors be liable for any direct, -// indirect, incidental, special, exemplary, or consequential damages -// (including, but not limited to, procurement of substitute goods or services; -// loss of use, data, or profits; or business interruption) however caused -// and on any theory of liability, whether in contract, strict liability, -// or tort (including negligence or otherwise) arising in any way out of -// the use of this software, even if advised of the possibility of such damage. -// -//M*/ - #include "opencv2/datasets/pd_inria.hpp" -#include - +#include +#include #include -#include -#include using namespace std; using namespace cv; @@ -55,6 +12,7 @@ int main(int argc, char *argv[]) const char *keys = "{ help h usage ? | | show this message }" "{ path p |true| path to dataset }"; + CommandLineParser parser(argc, argv, keys); string path(parser.get("path")); if (parser.has("help") || path=="true") @@ -66,28 +24,43 @@ int main(int argc, char *argv[]) Ptr dataset = PD_inria::create(); dataset->load(path); - cout << "train size: " << (unsigned int)dataset->getTrain().size() << endl; + size_t train_size = dataset->getTrain().size(); + size_t test_size = dataset->getTest().size(); + cout << "train size: " << train_size << endl; + cout << "test size: " << test_size << endl; - PD_inriaObj *example = static_cast(dataset->getTrain()[0].get()); - cout << "first train object: " << endl; - cout << "name: " << example->filename << endl; - - // image size - cout << "image size: " << endl; - cout << " - width: " << example->width << endl; - cout << " - height: " << example->height << endl; - cout << " - depth: " << example->depth << endl; - - // bounding boxes - for (unsigned int i = 0; i < (example->bndboxes).size(); i++) + for( size_t i = 0; i < train_size; i++ ) { - cout << "object " << i << endl; - int x = (example->bndboxes)[i].x; - int y = (example->bndboxes)[i].y; - cout << " - xmin = " << x << endl; - cout << " - ymin = " << y << endl; - cout << " - xmax = " << (example->bndboxes)[i].width + x << endl; - cout << " - ymax = " << (example->bndboxes)[i].height + y<< endl; + PD_inriaObj *example = static_cast(dataset->getTrain()[i].get()); + cout << "\ntrain object index: " << i << endl; + cout << "file name: " << example->filename << endl; + + // image size + cout << "image size: " << endl; + cout << " - width: " << example->width << endl; + cout << " - height: " << example->height << endl; + cout << " - depth: " << example->depth << endl; + + Mat img = imread( example->filename ); + + // bounding boxes + for ( size_t j = 0; j < example->bndboxes.size(); j++ ) + { + cout << "object " << j << endl; + int x = example->bndboxes[j].x; + int y = example->bndboxes[j].y; + cout << " - xmin = " << x << endl; + cout << " - ymin = " << y << endl; + cout << " - xmax = " << example->bndboxes[j].width + x << endl; + cout << " - ymax = " << example->bndboxes[j].height + y << endl; + rectangle( img, example->bndboxes[j], Scalar( 0, 0, 255 ), 2 ); + } + + imshow("INRIAPerson Dataset Train Images", img); + + cout << "\nPress a key to continue or ESC to exit." << endl; + int key = waitKey(); + if( key == 27 ) break; } return 0;