apps/haartraining/createsamples.cpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                        Intel License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000, Intel Corporation, all rights reserved.
  14 // Third party copyrights are property of their respective owners.
  15 //
  16 // Redistribution and use in source and binary forms, with or without modification,
  17 // are permitted provided that the following conditions are met:
  18 //
  19 //   * Redistribution's of source code must retain the above copyright notice,
  20 //     this list of conditions and the following disclaimer.
  21 //
  22 //   * Redistribution's in binary form must reproduce the above copyright notice,
  23 //     this list of conditions and the following disclaimer in the documentation
  24 //     and/or other materials provided with the distribution.
  25 //
  26 //   * The name of Intel Corporation may not be used to endorse or promote products
  27 //     derived from this software without specific prior written permission.
  28 //
  29 // This software is provided by the copyright holders and contributors "as is" and
  30 // any express or implied warranties, including, but not limited to, the implied
  31 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  32 // In no event shall the Intel Corporation or contributors be liable for any direct,
  33 // indirect, incidental, special, exemplary, or consequential damages
  34 // (including, but not limited to, procurement of substitute goods or services;
  35 // loss of use, data, or profits; or business interruption) however caused
  36 // and on any theory of liability, whether in contract, strict liability,
  37 // or tort (including negligence or otherwise) arising in any way out of
  38 // the use of this software, even if advised of the possibility of such damage.
  39 //
  40 //M*/
  41
  42 /*
  43  * createsamples.cpp
  44  *
  45  * Create test/training samples
  46  */
  47
  48 #include <cstdio>
  49 #include <cstring>
  50 #include <cstdlib>
  51 #include <cmath>
  52 #include <ctime>
  53
  54 using namespace std;
  55
  56 #include "cvhaartraining.h"
  57
  58 int main( int argc, char* argv[] )
  59 {
  60     int i = 0;
  61     char* nullname   = (char*)"(NULL)";
  62     char* vecname    = NULL; /* .vec file name */
  63     char* infoname   = NULL; /* file name with marked up image descriptions */
  64     char* imagename  = NULL; /* single sample image */
  65     char* bgfilename = NULL; /* background */
  66     int num = 1000;
  67     int bgcolor = 0;
  68     int bgthreshold = 80;
  69     int invert = 0;
  70     int maxintensitydev = 40;
  71     double maxxangle = 1.1;
  72     double maxyangle = 1.1;
  73     double maxzangle = 0.5;
  74     int showsamples = 0;
  75     /* the samples are adjusted to this scale in the sample preview window */
  76     double scale = 4.0;
  77     int width  = 24;
  78     int height = 24;
  79
  80     srand((unsigned int)time(0));
  81
  82     if( argc == 1 )
  83     {
  84         printf( "Usage: %s\n  [-info <collection_file_name>]\n"
  85                 "  [-img <image_file_name>]\n"
  86                 "  [-vec <vec_file_name>]\n"
  87                 "  [-bg <background_file_name>]\n  [-num <number_of_samples = %d>]\n"
  88                 "  [-bgcolor <background_color = %d>]\n"
  89                 "  [-inv] [-randinv] [-bgthresh <background_color_threshold = %d>]\n"
  90                 "  [-maxidev <max_intensity_deviation = %d>]\n"
  91                 "  [-maxxangle <max_x_rotation_angle = %f>]\n"
  92                 "  [-maxyangle <max_y_rotation_angle = %f>]\n"
  93                 "  [-maxzangle <max_z_rotation_angle = %f>]\n"
  94                 "  [-show [<scale = %f>]]\n"
  95                 "  [-w <sample_width = %d>]\n  [-h <sample_height = %d>]\n",
  96                 argv[0], num, bgcolor, bgthreshold, maxintensitydev,
  97                 maxxangle, maxyangle, maxzangle, scale, width, height );
  98
  99         return 0;
 100     }
 101
 102     for( i = 1; i < argc; ++i )
 103     {
 104         if( !strcmp( argv[i], "-info" ) )
 105         {
 106             infoname = argv[++i];
 107         }
 108         else if( !strcmp( argv[i], "-img" ) )
 109         {
 110             imagename = argv[++i];
 111         }
 112         else if( !strcmp( argv[i], "-vec" ) )
 113         {
 114             vecname = argv[++i];
 115         }
 116         else if( !strcmp( argv[i], "-bg" ) )
 117         {
 118             bgfilename = argv[++i];
 119         }
 120         else if( !strcmp( argv[i], "-num" ) )
 121         {
 122             num = atoi( argv[++i] );
 123         }
 124         else if( !strcmp( argv[i], "-bgcolor" ) )
 125         {
 126             bgcolor = atoi( argv[++i] );
 127         }
 128         else if( !strcmp( argv[i], "-bgthresh" ) )
 129         {
 130             bgthreshold = atoi( argv[++i] );
 131         }
 132         else if( !strcmp( argv[i], "-inv" ) )
 133         {
 134             invert = 1;
 135         }
 136         else if( !strcmp( argv[i], "-randinv" ) )
 137         {
 138             invert = CV_RANDOM_INVERT;
 139         }
 140         else if( !strcmp( argv[i], "-maxidev" ) )
 141         {
 142             maxintensitydev = atoi( argv[++i] );
 143         }
 144         else if( !strcmp( argv[i], "-maxxangle" ) )
 145         {
 146             maxxangle = atof( argv[++i] );
 147         }
 148         else if( !strcmp( argv[i], "-maxyangle" ) )
 149         {
 150             maxyangle = atof( argv[++i] );
 151         }
 152         else if( !strcmp( argv[i], "-maxzangle" ) )
 153         {
 154             maxzangle = atof( argv[++i] );
 155         }
 156         else if( !strcmp( argv[i], "-show" ) )
 157         {
 158             showsamples = 1;
 159             if( i+1 < argc && strlen( argv[i+1] ) > 0 && argv[i+1][0] != '-' )
 160             {
 161                 double d;
 162                 d = strtod( argv[i+1], 0 );
 163                 if( d != -HUGE_VAL && d != HUGE_VAL && d > 0 ) scale = d;
 164                 ++i;
 165             }
 166         }
 167         else if( !strcmp( argv[i], "-w" ) )
 168         {
 169             width = atoi( argv[++i] );
 170         }
 171         else if( !strcmp( argv[i], "-h" ) )
 172         {
 173             height = atoi( argv[++i] );
 174         }
 175     }
 176
 177     printf( "Info file name: %s\n", ((infoname == NULL) ?   nullname : infoname ) );
 178     printf( "Img file name: %s\n",  ((imagename == NULL) ?  nullname : imagename ) );
 179     printf( "Vec file name: %s\n",  ((vecname == NULL) ?    nullname : vecname ) );
 180     printf( "BG  file name: %s\n",  ((bgfilename == NULL) ? nullname : bgfilename ) );
 181     printf( "Num: %d\n", num );
 182     printf( "BG color: %d\n", bgcolor );
 183     printf( "BG threshold: %d\n", bgthreshold );
 184     printf( "Invert: %s\n", (invert == CV_RANDOM_INVERT) ? "RANDOM"
 185                                 : ( (invert) ? "TRUE" : "FALSE" ) );
 186     printf( "Max intensity deviation: %d\n", maxintensitydev );
 187     printf( "Max x angle: %g\n", maxxangle );
 188     printf( "Max y angle: %g\n", maxyangle );
 189     printf( "Max z angle: %g\n", maxzangle );
 190     printf( "Show samples: %s\n", (showsamples) ? "TRUE" : "FALSE" );
 191     if( showsamples )
 192     {
 193         printf( "Scale: %g\n", scale );
 194     }
 195     printf( "Width: %d\n", width );
 196     printf( "Height: %d\n", height );
 197
 198     /* determine action */
 199     if( imagename && vecname )
 200     {
 201         printf( "Create training samples from single image applying distortions...\n" );
 202
 203         cvCreateTrainingSamples( vecname, imagename, bgcolor, bgthreshold, bgfilename,
 204                                  num, invert, maxintensitydev,
 205                                  maxxangle, maxyangle, maxzangle,
 206                                  showsamples, width, height );
 207
 208         printf( "Done\n" );
 209     }
 210     else if( imagename && bgfilename && infoname )
 211     {
 212         printf( "Create test samples from single image applying distortions...\n" );
 213
 214         cvCreateTestSamples( infoname, imagename, bgcolor, bgthreshold, bgfilename, num,
 215             invert, maxintensitydev,
 216             maxxangle, maxyangle, maxzangle, showsamples, width, height );
 217
 218         printf( "Done\n" );
 219     }
 220     else if( infoname && vecname )
 221     {
 222         int total;
 223
 224         printf( "Create training samples from images collection...\n" );
 225
 226         total = cvCreateTrainingSamplesFromInfo( infoname, vecname, num, showsamples,
 227                                                  width, height );
 228
 229         printf( "Done. Created %d samples\n", total );
 230     }
 231     else if( vecname )
 232     {
 233         printf( "View samples from vec file (press ESC to exit)...\n" );
 234
 235         cvShowVecSamples( vecname, width, height, scale );
 236
 237         printf( "Done\n" );
 238     }
 239     else
 240     {
 241         printf( "Nothing to do\n" );
 242     }
 243
 244     return 0;
 245 }