txt2bin.cpp

基于VC开发的神经网络工具箱

#include <iostream>
#include "../include/annie.h"

using namespace std;
using namespace annie;

#define MAX_FILENAME_SIZE 255

int main(int argc, char *argv[])
{
	cout<<"Usage: "<<argv[0]<<" <text TrainingSet> <binary TrainingSet>"<<endl;
	char inFile[MAX_FILENAME_SIZE];
	char outFile[MAX_FILENAME_SIZE];

	if (argc>1)
	{
		strcpy(inFile,argv[1]);
		cout<<"Enter input filename (text training set)   : "<<inFile<<endl;
	}
	else
	{
        cout<<"Enter input filename (text training set)   : "; cin>>inFile;
	}

	if (argc>2)
	{
		strcpy(outFile,argv[2]);
		cout<<"Enter output filename (binary training set): "<<outFile<<endl;
	}
	else
	{
		cout<<"Enter output filename (binary training set): "; cin>>outFile;
	}

	TrainingSet *T;
	try
	{
		T = new TrainingSet(inFile,TEXT_FILE);
		cout<<"TrainingSet of size "<<T->getSize()<<" read."<<endl;
		T->save(outFile,BINARY_FILE);
	}
	catch (Exception e)
	{
		cerr<<e.what()<<endl;
		exit(-1);
	}
	cout<<"Conversion completed successfully"<<endl;
	delete T;

	return 0;
}