mirror of https://github.com/opencv/opencv.git
vbystricky
10 years ago
parent
0a4c616d43
commit
c48f30d4ec
5 changed files with 2090 additions and 0 deletions
@ -0,0 +1,39 @@ |
||||
set(OPENCV_CREATESAMPLES_DEPS opencv_core opencv_imgproc opencv_objdetect opencv_imgcodecs opencv_highgui opencv_calib3d opencv_features2d opencv_videoio) |
||||
ocv_check_dependencies(${OPENCV_CREATESAMPLES_DEPS}) |
||||
|
||||
if(NOT OCV_DEPENDENCIES_FOUND) |
||||
return() |
||||
endif() |
||||
|
||||
project(createsamples) |
||||
set(the_target opencv_createsamples) |
||||
|
||||
ocv_target_include_directories(${the_target} PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" "${OpenCV_SOURCE_DIR}/include/opencv") |
||||
ocv_target_include_modules(${the_target} ${OPENCV_CREATESAMPLES_DEPS}) |
||||
|
||||
file(GLOB SRCS *.cpp) |
||||
file(GLOB HDRS *.h*) |
||||
|
||||
set(createsamples_files ${SRCS} ${HDRS}) |
||||
|
||||
ocv_add_executable(${the_target} ${createsamples_files}) |
||||
ocv_target_link_libraries(${the_target} ${OPENCV_CREATESAMPLES_DEPS}) |
||||
|
||||
set_target_properties(${the_target} PROPERTIES |
||||
DEBUG_POSTFIX "${OPENCV_DEBUG_POSTFIX}" |
||||
ARCHIVE_OUTPUT_DIRECTORY ${LIBRARY_OUTPUT_PATH} |
||||
RUNTIME_OUTPUT_DIRECTORY ${EXECUTABLE_OUTPUT_PATH} |
||||
INSTALL_NAME_DIR lib |
||||
OUTPUT_NAME "opencv_createsamples") |
||||
|
||||
if(ENABLE_SOLUTION_FOLDERS) |
||||
set_target_properties(${the_target} PROPERTIES FOLDER "applications") |
||||
endif() |
||||
|
||||
if(INSTALL_CREATE_DISTRIB) |
||||
if(BUILD_SHARED_LIBS) |
||||
install(TARGETS ${the_target} RUNTIME DESTINATION ${OPENCV_BIN_INSTALL_PATH} CONFIGURATIONS Release COMPONENT dev) |
||||
endif() |
||||
else() |
||||
install(TARGETS ${the_target} RUNTIME DESTINATION ${OPENCV_BIN_INSTALL_PATH} COMPONENT dev) |
||||
endif() |
||||
@ -0,0 +1,245 @@ |
||||
/*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.
|
||||
//
|
||||
//
|
||||
// Intel License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000, Intel Corporation, 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 Intel Corporation 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 Intel Corporation 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*/
|
||||
|
||||
/*
|
||||
* createsamples.cpp |
||||
* |
||||
* Create test/training samples |
||||
*/ |
||||
|
||||
#include <cstdio> |
||||
#include <cstring> |
||||
#include <cstdlib> |
||||
#include <cmath> |
||||
#include <ctime> |
||||
|
||||
using namespace std; |
||||
|
||||
#include "utility.hpp" |
||||
|
||||
int main( int argc, char* argv[] ) |
||||
{ |
||||
int i = 0; |
||||
char* nullname = (char*)"(NULL)"; |
||||
char* vecname = NULL; /* .vec file name */ |
||||
char* infoname = NULL; /* file name with marked up image descriptions */ |
||||
char* imagename = NULL; /* single sample image */ |
||||
char* bgfilename = NULL; /* background */ |
||||
int num = 1000; |
||||
int bgcolor = 0; |
||||
int bgthreshold = 80; |
||||
int invert = 0; |
||||
int maxintensitydev = 40; |
||||
double maxxangle = 1.1; |
||||
double maxyangle = 1.1; |
||||
double maxzangle = 0.5; |
||||
int showsamples = 0; |
||||
/* the samples are adjusted to this scale in the sample preview window */ |
||||
double scale = 4.0; |
||||
int width = 24; |
||||
int height = 24; |
||||
|
||||
srand((unsigned int)time(0)); |
||||
|
||||
if( argc == 1 ) |
||||
{ |
||||
printf( "Usage: %s\n [-info <collection_file_name>]\n" |
||||
" [-img <image_file_name>]\n" |
||||
" [-vec <vec_file_name>]\n" |
||||
" [-bg <background_file_name>]\n [-num <number_of_samples = %d>]\n" |
||||
" [-bgcolor <background_color = %d>]\n" |
||||
" [-inv] [-randinv] [-bgthresh <background_color_threshold = %d>]\n" |
||||
" [-maxidev <max_intensity_deviation = %d>]\n" |
||||
" [-maxxangle <max_x_rotation_angle = %f>]\n" |
||||
" [-maxyangle <max_y_rotation_angle = %f>]\n" |
||||
" [-maxzangle <max_z_rotation_angle = %f>]\n" |
||||
" [-show [<scale = %f>]]\n" |
||||
" [-w <sample_width = %d>]\n [-h <sample_height = %d>]\n", |
||||
argv[0], num, bgcolor, bgthreshold, maxintensitydev, |
||||
maxxangle, maxyangle, maxzangle, scale, width, height ); |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
for( i = 1; i < argc; ++i ) |
||||
{ |
||||
if( !strcmp( argv[i], "-info" ) ) |
||||
{ |
||||
infoname = argv[++i]; |
||||
} |
||||
else if( !strcmp( argv[i], "-img" ) ) |
||||
{ |
||||
imagename = argv[++i]; |
||||
} |
||||
else if( !strcmp( argv[i], "-vec" ) ) |
||||
{ |
||||
vecname = argv[++i]; |
||||
} |
||||
else if( !strcmp( argv[i], "-bg" ) ) |
||||
{ |
||||
bgfilename = argv[++i]; |
||||
} |
||||
else if( !strcmp( argv[i], "-num" ) ) |
||||
{ |
||||
num = atoi( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-bgcolor" ) ) |
||||
{ |
||||
bgcolor = atoi( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-bgthresh" ) ) |
||||
{ |
||||
bgthreshold = atoi( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-inv" ) ) |
||||
{ |
||||
invert = 1; |
||||
} |
||||
else if( !strcmp( argv[i], "-randinv" ) ) |
||||
{ |
||||
invert = CV_RANDOM_INVERT; |
||||
} |
||||
else if( !strcmp( argv[i], "-maxidev" ) ) |
||||
{ |
||||
maxintensitydev = atoi( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-maxxangle" ) ) |
||||
{ |
||||
maxxangle = atof( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-maxyangle" ) ) |
||||
{ |
||||
maxyangle = atof( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-maxzangle" ) ) |
||||
{ |
||||
maxzangle = atof( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-show" ) ) |
||||
{ |
||||
showsamples = 1; |
||||
if( i+1 < argc && strlen( argv[i+1] ) > 0 && argv[i+1][0] != '-' ) |
||||
{ |
||||
double d; |
||||
d = strtod( argv[i+1], 0 ); |
||||
if( d != -HUGE_VAL && d != HUGE_VAL && d > 0 ) scale = d; |
||||
++i; |
||||
} |
||||
} |
||||
else if( !strcmp( argv[i], "-w" ) ) |
||||
{ |
||||
width = atoi( argv[++i] ); |
||||
} |
||||
else if( !strcmp( argv[i], "-h" ) ) |
||||
{ |
||||
height = atoi( argv[++i] ); |
||||
} |
||||
} |
||||
|
||||
printf( "Info file name: %s\n", ((infoname == NULL) ? nullname : infoname ) ); |
||||
printf( "Img file name: %s\n", ((imagename == NULL) ? nullname : imagename ) ); |
||||
printf( "Vec file name: %s\n", ((vecname == NULL) ? nullname : vecname ) ); |
||||
printf( "BG file name: %s\n", ((bgfilename == NULL) ? nullname : bgfilename ) ); |
||||
printf( "Num: %d\n", num ); |
||||
printf( "BG color: %d\n", bgcolor ); |
||||
printf( "BG threshold: %d\n", bgthreshold ); |
||||
printf( "Invert: %s\n", (invert == CV_RANDOM_INVERT) ? "RANDOM" |
||||
: ( (invert) ? "TRUE" : "FALSE" ) ); |
||||
printf( "Max intensity deviation: %d\n", maxintensitydev ); |
||||
printf( "Max x angle: %g\n", maxxangle ); |
||||
printf( "Max y angle: %g\n", maxyangle ); |
||||
printf( "Max z angle: %g\n", maxzangle ); |
||||
printf( "Show samples: %s\n", (showsamples) ? "TRUE" : "FALSE" ); |
||||
if( showsamples ) |
||||
{ |
||||
printf( "Scale: %g\n", scale ); |
||||
} |
||||
printf( "Width: %d\n", width ); |
||||
printf( "Height: %d\n", height ); |
||||
|
||||
/* determine action */ |
||||
if( imagename && vecname ) |
||||
{ |
||||
printf( "Create training samples from single image applying distortions...\n" ); |
||||
|
||||
cvCreateTrainingSamples( vecname, imagename, bgcolor, bgthreshold, bgfilename, |
||||
num, invert, maxintensitydev, |
||||
maxxangle, maxyangle, maxzangle, |
||||
showsamples, width, height ); |
||||
|
||||
printf( "Done\n" ); |
||||
} |
||||
else if( imagename && bgfilename && infoname ) |
||||
{ |
||||
printf( "Create test samples from single image applying distortions...\n" ); |
||||
|
||||
cvCreateTestSamples( infoname, imagename, bgcolor, bgthreshold, bgfilename, num, |
||||
invert, maxintensitydev, |
||||
maxxangle, maxyangle, maxzangle, showsamples, width, height ); |
||||
|
||||
printf( "Done\n" ); |
||||
} |
||||
else if( infoname && vecname ) |
||||
{ |
||||
int total; |
||||
|
||||
printf( "Create training samples from images collection...\n" ); |
||||
|
||||
total = cvCreateTrainingSamplesFromInfo( infoname, vecname, num, showsamples, |
||||
width, height ); |
||||
|
||||
printf( "Done. Created %d samples\n", total ); |
||||
} |
||||
else if( vecname ) |
||||
{ |
||||
printf( "View samples from vec file (press ESC to exit)...\n" ); |
||||
|
||||
cvShowVecSamples( vecname, width, height, scale ); |
||||
|
||||
printf( "Done\n" ); |
||||
} |
||||
else |
||||
{ |
||||
printf( "Nothing to do\n" ); |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,124 @@ |
||||
/*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.
|
||||
//
|
||||
//
|
||||
// Intel License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000, Intel Corporation, 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 Intel Corporation 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 Intel Corporation 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*/
|
||||
|
||||
#ifndef __CREATESAMPLES_UTILITY_HPP__ |
||||
#define __CREATESAMPLES_UTILITY_HPP__ |
||||
|
||||
#define CV_VERBOSE 1 |
||||
|
||||
/*
|
||||
* cvCreateTrainingSamples |
||||
* |
||||
* Create training samples applying random distortions to sample image and |
||||
* store them in .vec file |
||||
* |
||||
* filename - .vec file name |
||||
* imgfilename - sample image file name |
||||
* bgcolor - background color for sample image |
||||
* bgthreshold - background color threshold. Pixels those colors are in range |
||||
* [bgcolor-bgthreshold, bgcolor+bgthreshold] are considered as transparent |
||||
* bgfilename - background description file name. If not NULL samples |
||||
* will be put on arbitrary background |
||||
* count - desired number of samples |
||||
* invert - if not 0 sample foreground pixels will be inverted |
||||
* if invert == CV_RANDOM_INVERT then samples will be inverted randomly |
||||
* maxintensitydev - desired max intensity deviation of foreground samples pixels |
||||
* maxxangle - max rotation angles |
||||
* maxyangle |
||||
* maxzangle |
||||
* showsamples - if not 0 samples will be shown |
||||
* winwidth - desired samples width |
||||
* winheight - desired samples height |
||||
*/ |
||||
#define CV_RANDOM_INVERT 0x7FFFFFFF |
||||
|
||||
void cvCreateTrainingSamples( const char* filename, |
||||
const char* imgfilename, int bgcolor, int bgthreshold, |
||||
const char* bgfilename, int count, |
||||
int invert = 0, int maxintensitydev = 40, |
||||
double maxxangle = 1.1, |
||||
double maxyangle = 1.1, |
||||
double maxzangle = 0.5, |
||||
int showsamples = 0, |
||||
int winwidth = 24, int winheight = 24 ); |
||||
|
||||
void cvCreateTestSamples( const char* infoname, |
||||
const char* imgfilename, int bgcolor, int bgthreshold, |
||||
const char* bgfilename, int count, |
||||
int invert, int maxintensitydev, |
||||
double maxxangle, double maxyangle, double maxzangle, |
||||
int showsamples, |
||||
int winwidth, int winheight ); |
||||
|
||||
/*
|
||||
* cvCreateTrainingSamplesFromInfo |
||||
* |
||||
* Create training samples from a set of marked up images and store them into .vec file |
||||
* infoname - file in which marked up image descriptions are stored |
||||
* num - desired number of samples |
||||
* showsamples - if not 0 samples will be shown |
||||
* winwidth - sample width |
||||
* winheight - sample height |
||||
* |
||||
* Return number of successfully created samples |
||||
*/ |
||||
int cvCreateTrainingSamplesFromInfo( const char* infoname, const char* vecfilename, |
||||
int num, |
||||
int showsamples, |
||||
int winwidth, int winheight ); |
||||
|
||||
/*
|
||||
* cvShowVecSamples |
||||
* |
||||
* Shows samples stored in .vec file |
||||
* |
||||
* filename |
||||
* .vec file name |
||||
* winwidth |
||||
* sample width |
||||
* winheight |
||||
* sample height |
||||
* scale |
||||
* the scale each sample is adjusted to |
||||
*/ |
||||
void cvShowVecSamples( const char* filename, int winwidth, int winheight, double scale ); |
||||
|
||||
#endif //__CREATESAMPLES_UTILITY_HPP__
|
||||
Loading…
Reference in new issue