/*
* Find the closest match in db, return as element of matches
*/
void SemanticDescriptor::findClosestMatch(Database & db, Matches & matches)
{
SemanticDescriptor bestMatch = (*(db.dDB.begin())).first;
//matches.insert((*(db.dDB.begin())).second);
SemanticDescriptor compareID;
int matchStrength;
map<SemanticDescriptor, string>::iterator iter;
for (iter = db.dDB.begin(); iter != db.dDB.end(); ++iter) {
compareID = iter->first;
matchStrength = closerMatch(compareID, bestMatch, db);
/* If compareID is a better match, reset matches */
if (matchStrength > 0)
if (compareID.coverage(*this, db) >= .5) {
matches.clear();
matches.insert(iter->second);
bestMatch = compareID;
}
/* If compareID is an equal match, then add it to the set of matches */
if (matchStrength == 0)
if (compareID.coverage(*this, db) >= .5)
matches.insert(iter->second);
}
}
/*
* Find exact matches for the semantic descriptor in the database
*/
bool SemanticDescriptor::findExactMatches(Database & db, Matches & matches)
{
SemanticDescriptor compareID;
map<SemanticDescriptor, string>::iterator iter;
for (iter = db.dDB.begin(); iter != db.dDB.end(); ++iter) {
compareID = iter->first;
if (equals(compareID, db)) {
matches.insert(iter->second);
}
}
return matches.size();
}
int main(int ac, char* av[])
{
cout << "InfinitePanorama. v0.2 -- (C) 2012 by Wilston Oreo." << endl;
stringstream descStr;
descStr << "Allowed options:" << endl;
//descStr << "Panorama mode (interactive):" << endl;
//descStr << "\tinfinitepanorama -P -l left.dat -r right.dat" << endl << endl;
descStr << "Panorama mode (generate output file):" << endl;
descStr << "\tinfinitepanorama -P -l left.dat -r right.dat -n 10 -h 1024 -o output.png" << endl << endl;
descStr << "Database mode (generate database from directory):" << endl;
descStr << "\tinfinitepanorama -D -i inputdir -l left.dat -r right.dat" << endl;
descStr << "Database mode (generate database from filelist):" << endl;
descStr << "\tinfinitepanorama -D -f filelist -l left.dat -r right.dat" << endl;
descStr << "Preprocessing mode (generate database from filelist):" << endl;
descStr << "\tinfinitepanorama -S -f filelist -i input.jpg -l left.dat -r right.dat" << endl;
po::options_description desc(descStr.str());
string inputDir, fileList, databaseFileLeft, databaseFileRight;
string outputImageFile, configFile;
int width = 10240;
int height = 768;
int gistCount = 10000,histLargeCount = 100, histSmallCount = 0, thumbCount = 0;
desc.add_options()
("help", "Display help message.")
("database,D", "Database mode")
("panorama,P", "Panorama mode")
("preprocess,S","Preprocessing mode")
("config,c", po::value<string>(&configFile), "Config file")
("inputdir,i", po::value<string>(&inputDir), "Input directory")
("filelist,f", po::value<string>(&fileList), "File list or data base list")
("left,l", po::value<string>(&databaseFileLeft), "Database for left part of images")
("right,r", po::value<string>(&databaseFileRight),"Database for right part of images")
("output,o", po::value<string>(&outputImageFile), "Output image file")
("width,w", po::value<int>(&width), "Panorama width. Default: 10240")
("height,h", po::value<int>(&height), "Panorama height. Default: 768")
("gist", po::value<int>(&gistCount), "Gist count")
("histlarge", po::value<int>(&histLargeCount), "Histogram (large) count")
("histsmall", po::value<int>(&histSmallCount), "Histogram (small) count")
("thumb", po::value<int>(&thumbCount), "Thumbnail count")
("linear", "Enable linear blending (default)")
("poisson", "Enable poisson blending")
;
po::variables_map vm;
po::store(po::parse_command_line(ac, av, desc), vm);
po::notify(vm);
#define V vm.count
if (V("help")) { cout << desc << endl; return 1; }
if (V("config")) loadConfig(configFile);
if (V("gist")) config.set("FILTER_GIST_MATCHES",gistCount);
if (V("histsmall")) config.set("FILTER_HISTSMALL_MATCHES",histSmallCount);
if (V("histlarge")) config.set("FILTER_HISTLARGE_MATCHES",histLargeCount);
if (V("thumb")) config.set("FILTER_THUMBNAIL_MATCHES",thumbCount);
Panorama pan(width,height,&config);
config.print();
Panorama::BlendingMode blend = Panorama::BLEND_LINEAR;
if (V("poisson")) blend = Panorama::BLEND_POISSON;
if (V("preprocess") && V("left") && V("right") && V("inputdir"))
{
Image image(inputDir);
float border = 1.0f/6.0f;
Descriptor leftDesc(image,int(image.columns()*border),image.rows(),0,0);
Descriptor rightDesc(image,int(image.columns()*border),image.rows(),int((1.0f-border)*image.columns()),0);
LOG->level(2);
config.set("FILTER_GIST_MATCHES",5000);
config.set("FILTER_HISTSMALL_MATCHES",0);
config.set("FILTER_HISTLARGE_MATCHES",0);
config.set("FILTER_THUMBNAIL_MATCHES",0);
Database leftFinal(&config), rightFinal(&config);
vector< left_right > files = loadDatabaseList(fileList);
BOOST_FOREACH( left_right& file, files)
{
Statistics statistics(0,true);
DescriptorFilter filter(&config,&statistics);
LOG_MSG << fmt("Reading databases '%' (left) and '%' (right)...") % file.first % file.second;
Database leftTmp(file.first), rightTmp(file.second);
Descriptors leftDescs = leftTmp.descriptors(), rightDescs = rightTmp.descriptors();
Matches leftMatches = filter.getMatches(leftDesc,leftDescs);
BOOST_FOREACH ( const Match& m, leftMatches)
{
if (m.desc)
{
statistics.exclude(leftDescs[m.desc->index()]);
statistics.exclude(rightDescs[m.desc->index()]);
}
}
Matches rightMatches = filter.getMatches(rightDesc,rightDescs);
Matches matches = leftMatches;
BOOST_FOREACH ( const Match& m, rightMatches )
matches.insert(m);
BOOST_FOREACH ( const Match& m, matches)
{
if (m.desc)
{
Descriptor* newLeft = new Descriptor(leftTmp[m.desc->index()]);
Descriptor* newRight = new Descriptor(rightTmp[m.desc->index()]);
leftFinal.push_back(newLeft);
rightFinal.push_back(newRight);
}
}
LOG_MSG << fmt("Left database contains %, right database contains %") % leftFinal.size() % rightFinal.size();
leftTmp.clear();
rightTmp.clear();
}