void Accessor::Load(
	lpxmlnode						pNode)
{
	m_source	= GetAttributeStringValue("source", pNode);
	m_count		= GetAttributeIntegerValue("count", pNode);
	m_stride	= GetAttributeIntegerValue("stride", pNode);

	uint i=0;
	lpxmlnode pCurrNode = pNode->first_node();
	do {
		std::string Name = pCurrNode->name();
		if(Name == "param") {
			Param prm;
			prm.Load(pCurrNode);
			m_param.push_back(prm);
		} 
		pCurrNode = pCurrNode->next_sibling();
	}	while(pCurrNode != NULL);
}
Beispiel #2
0
int main(int argc, char* argv[])
{
	OPTIONS opt;
	if(!LoadConfiguration(argc, argv, opt))
		return 1;

	clock_t time_b, time_e;
	time_b = clock();

	try {
		// load input lists
		CFileList files("Files");
		CFileList **fileLists;

		loadFileLists(files, opt);

		// load GMM
		GMModel GMM;
		if (GMM.Load(opt.GMM_file.c_str(), opt.load_txt) != 0) {
			error_report = string("GMM model [") + opt.GMM_file + "] not found";
			throw runtime_error(error_report.c_str());
		}
		unsigned int M = GMM.GetNumberOfMixtures();
		unsigned int D = GMM.GetDimension();

		// prepare estimator
		GMMStatsEstimator<realT> *estimator;
		if(opt.SSEacc) {
			estimator = new GMMStatsEstimator_SSE <realT>;
		}
#ifdef _CUDA
		else if(opt.CUDAacc) {
			estimator = new GMMStatsEstimator_CUDA <realT> (opt.cudaGPUid);
		}
#endif
		else {
			estimator = new GMMStatsEstimator <realT>;
		}

		if(opt.verbosity > 1)
			cout << " [" << estimator->getEstimationType().c_str() << " Estimation ON]" << endl;

		estimator->insertModel(GMM);
		estimator->_numThreads = opt.numThreads;
		estimator->_verbosity = opt.verbosity;

		char *filename;
		string outFileName;

		if(opt.verbosity > 1)
			cout << "Log-likes are going to be computed [#files = " << files.ListLength() << "]" << endl;

		float *outLLs = NULL;
		unsigned int outLLs_size = 0;

		Param param;
		files.Rewind();
		while(files.GetItemName(&filename)) {		
			
			if(param.Load(filename, opt.load_type, opt.dwnsmp) != 0) {
				cerr << "WARNING: Param file [" << filename << "] not found -> skipped!" << endl;
				continue;
			}

			unsigned int NSamples = (unsigned int) param.GetNumberOfVectors();
			unsigned int prmdim = (unsigned int) param.GetVectorDim();

			if(NSamples < 1 && opt.verbosity > 1) {
				cerr << "WARNING: Param file [" << filename << "] empty" << endl;
				continue;
			}

			if (NSamples > outLLs_size) {
				delete[] outLLs;
				outLLs = new float [NSamples];
				outLLs_size = NSamples;
			}
			estimator->compVecsLogLikeMT(*param.GetVectors(), NSamples, prmdim, outLLs);

			if(estimator->_fNumStabilityTroubles) {
				cerr << "WARNING: numerical stability troubles detected [file> " << filename << "]" << endl;
				estimator->_fNumStabilityTroubles = false;
			}

			if (opt.forEachVec) {
				string foo(filename);
				GetNewPath(outFileName, foo, opt.outputDir, opt.outExt);
				saveMat<float> (outFileName.c_str(), opt.save_txt, outLLs, 1, NSamples);
			}
			else {
				float meanLL = 0.0f;
				for (unsigned int n = 0; n < NSamples; n++)
					meanLL += outLLs[n];
				meanLL /= NSamples;
				cout << "\r" << filename << "\t" << meanLL << "\n";
			}
		}

		delete estimator;
		delete[] outLLs;
	}
	catch (bad_alloc& ba) {
		cerr << "Out of memory!" << endl;
		cerr << ba.what() << endl;
	}
	catch (exception& e) {
		cerr << e.what() << endl;		
	}
	catch (...) {
		cerr << "Unknown exception caught!" << endl << "Please, contact the developer." << endl;
	}

	time_e = clock();
	if(opt.verbosity > 1)
		cout << endl << "[Processing time = " << (time_e - time_b)/(double) CLOCKS_PER_SEC << " seconds]" << endl;

	return 0;
}