generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added number of bands estimation in opencv_stitching

Browse Source
This commit is contained in:
Alexey Spizhevoy 2011-05-20 07:44:37 +00:00
parent 767a6e8e78
commit 5b50d63754
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/stitching/blenders.hpp
View File

@ -41,6 +41,9 @@ class MultiBandBlender : public Blender
public:
MultiBandBlender(int num_bands = 7) : num_bands_(num_bands) {}
int numBands() const { return num_bands_; }
void setNumBands(int val) { num_bands_ = val; }
private:
cv::Point blend(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
cv::Mat& dst, cv::Mat& dst_mask);

7
modules/stitching/main.cpp
View File

@ -347,8 +347,13 @@ int main(int argc, char* argv[])
t = getTickCount();
LOGLN("Blending images...");
Mat result, result_mask;
Ptr<Blender> blender = Blender::createDefault(blend_type);
if (blend_type == Blender::MULTI_BAND)
// Ensure last pyramid layer area is about 1 pix
dynamic_cast<MultiBandBlender*>((Blender*)(blender))
->setNumBands(static_cast<int>(ceil(log(static_cast<double>(images_f[0].size().area()))
/ log(4.0))));
Mat result, result_mask;
(*blender)(images_f, corners, masks_warped, result, result_mask);
LOGLN("Blending images, time: " << ((getTickCount() - t) / getTickFrequency()) << " sec");