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

fixed save/load of dot; added data for the sample

Browse Source
This commit is contained in:
Maria Dimashova
2011-05-10 13:30:58 +00:00
parent 1c9f4e7ca2
commit 0f6b8002dc
9 changed files with 44 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
samples/cpp/dot.cpp
View File

@@ -41,15 +41,15 @@ int main( int argc, char **argv )
{
if( argc != 1 && argc != 3 )
{
cout << "Format: base; " << endl << "or without arguments to use default data" << endl;
cout << "Format: train_data test_data; " << endl << "or without arguments to use default data" << endl;
return -1;
}
string baseDirName, testDirName;
if( argc == 1 )
{
baseDirName = "../MultiScaleDOT/Data/train/";
testDirName = "../MultiScaleDOT/Data/test/";
baseDirName = "../../opencv/samples/cpp/dot_data/train/";
testDirName = "../../opencv/samples/cpp/dot_data/test/";
}
else
{
@@ -71,6 +71,9 @@ int main( int argc, char **argv )
DOTDetector dotDetector;
dotDetector.train( baseDirName, trainParams, true );
// dotDetector.save( "../../dot.xml.gz" );
// dotDetector.load( "../../dot.xml.gz" );
const vector<string>& objectClassNames = dotDetector.getObjectClassNames();
const vector<DOTDetector::DOTTemplate>& dotTemplates = dotDetector.getDOTTemplates();