made number of bands in blending parameter

This commit is contained in:
Alexey Spizhevoy 2011-05-21 14:44:37 +00:00
parent 97530caa53
commit 868035215f
2 changed files with 10 additions and 5 deletions

View File

@ -43,7 +43,7 @@ private:
class MultiBandBlender : public Blender class MultiBandBlender : public Blender
{ {
public: public:
MultiBandBlender(int num_bands = 7) { setNumBands(num_bands); } MultiBandBlender(int num_bands = 5) { setNumBands(num_bands); }
int numBands() const { return num_bands_; } int numBands() const { return num_bands_; }
void setNumBands(int val) { num_bands_ = val; } void setNumBands(int val) { num_bands_ = val; }

View File

@ -23,6 +23,7 @@ void printUsage()
<< "\t[--warp (plane|cylindrical|spherical)]\n" << "\t[--warp (plane|cylindrical|spherical)]\n"
<< "\t[--seam (no|voronoi|graphcut)]\n" << "\t[--seam (no|voronoi|graphcut)]\n"
<< "\t[--blend (no|feather|multiband)]\n" << "\t[--blend (no|feather|multiband)]\n"
<< "\t[--numbands <int>]\n"
<< "\t[--output <result_img>]\n\n"; << "\t[--output <result_img>]\n\n";
cout << "--matchconf\n" cout << "--matchconf\n"
<< "\tGood values are in [0.2, 0.8] range usually.\n\n"; << "\tGood values are in [0.2, 0.8] range usually.\n\n";
@ -45,6 +46,7 @@ bool user_match_conf = false;
float match_conf = 0.6f; float match_conf = 0.6f;
int seam_find_type = SeamFinder::VORONOI; int seam_find_type = SeamFinder::VORONOI;
int blend_type = Blender::MULTI_BAND; int blend_type = Blender::MULTI_BAND;
int numbands = 5;
string result_name = "result.png"; string result_name = "result.png";
int parseCmdArgs(int argc, char** argv) int parseCmdArgs(int argc, char** argv)
@ -173,6 +175,11 @@ int parseCmdArgs(int argc, char** argv)
} }
i++; i++;
} }
else if (string(argv[i]) == "--numbands")
{
numbands = atoi(argv[i + 1]);
i++;
}
else if (string(argv[i]) == "--output") else if (string(argv[i]) == "--output")
{ {
result_name = argv[i + 1]; result_name = argv[i + 1];
@ -410,14 +417,12 @@ int main(int argc, char* argv[])
if (static_cast<Blender*>(blender) == 0) if (static_cast<Blender*>(blender) == 0)
{ {
// Create blender
blender = Blender::createDefault(blend_type); blender = Blender::createDefault(blend_type);
if (blend_type == Blender::MULTI_BAND) if (blend_type == Blender::MULTI_BAND)
{ {
// Ensure last pyramid layer area is about 1 pix
MultiBandBlender* mb = dynamic_cast<MultiBandBlender*>(static_cast<Blender*>(blender)); MultiBandBlender* mb = dynamic_cast<MultiBandBlender*>(static_cast<Blender*>(blender));
mb->setNumBands(static_cast<int>(ceil(log(static_cast<double>(img_warped_f.size().area())) / log(4.0)))); mb->setNumBands(numbands);
} }
// Determine the final image size // Determine the final image size