CVRecoMain.cpp

#include "RecoManager.h"
#include <stdlib.h>
#include <iostream>

using namespace std;

#define NUMARGUMENTS 5
void Usage(char* argv[])
{
	cout << "Usage: " << argv[0] << " <Base Folder Path> <Name of Log File> <number of features to detect> <dictionary size (kmeans clusters> <0 for validation image prediction, 1 for test image prediction>" << endl;
	exit(-2);
}

int main(int argc, char* argv[])
{
	//Check usage
	if(argc != NUMARGUMENTS+1)
	{
		Usage(argv);
	}

	CRecoManager recoMgr;
	unsigned int numFeatures = (unsigned int)atoi(argv[3]);
	unsigned int dictionarySize = (unsigned int)atoi(argv[4]);
	unsigned int validationOrTest = (unsigned int)atoi(argv[5]);
	
	if(!recoMgr.Init(string (argv[1]), string(argv[2]), string(argv[3]), string(argv[4]), string(argv[5]))) //Initialize
	{
		return -1;
	}

	//Populate images
	cout << "Populating Image List..." << endl;
	recoMgr.PopulateImages(validationOrTest);

	cout << "Building Model..." << endl;
	if(!recoMgr.BuildModel(numFeatures, dictionarySize))
	{
		return -1;
	}
	if(!recoMgr.TestModel(numFeatures))
	{
		return -1;
	}

	return 0;
}