mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.4 KiB
72 lines
2.4 KiB
14 years ago
|
#ifndef _OPENCV_BLENDERS_HPP_
|
||
|
|
#define _OPENCV_BLENDERS_HPP_
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <opencv2/core/core.hpp>
|
||
|
|
|
||
|
|
// Simple blender which puts one image over another
|
||
|
|
class Blender
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
enum { NO, FEATHER, MULTI_BAND };
|
||
|
|
|
||
|
|
static cv::Ptr<Blender> createDefault(int type);
|
||
|
|
|
||
|
|
cv::Point operator ()(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
||
|
|
cv::Mat& dst);
|
||
|
|
cv::Point operator ()(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);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
virtual 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);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class FeatherBlender : public Blender
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
FeatherBlender(float sharpness = 0.02f) : sharpness_(sharpness) {}
|
||
|
|
|
||
|
|
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);
|
||
|
|
|
||
|
|
float sharpness_;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class MultiBandBlender : public Blender
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
MultiBandBlender(int num_bands = 10) : num_bands_(num_bands) {}
|
||
|
|
|
||
|
|
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);
|
||
|
|
|
||
|
|
int num_bands_;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
// Auxiliary functions
|
||
|
|
|
||
|
|
cv::Rect resultRoi(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners);
|
||
|
|
|
||
|
|
cv::Point computeResultMask(const std::vector<cv::Mat> &masks, const std::vector<cv::Point> &corners, cv::Mat &mask);
|
||
|
|
|
||
|
|
cv::Point blendLinear(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &weights,
|
||
|
|
cv::Mat& dst, cv::Mat& dst_weight);
|
||
|
|
|
||
|
|
void createWeightMap(const cv::Mat& mask, float sharpness, cv::Mat& weight);
|
||
|
|
|
||
|
|
void createGaussPyr(const cv::Mat& img, int num_layers, std::vector<cv::Mat>& pyr);
|
||
|
|
|
||
|
|
void createLaplacePyr(const std::vector<cv::Mat>& pyr_gauss, std::vector<cv::Mat>& pyr_laplace);
|
||
|
|
|
||
|
|
// Restores source image in-place. Result will be in pyr[0].
|
||
|
|
void restoreImageFromLaplacePyr(std::vector<cv::Mat>& pyr);
|
||
|
|
|
||
|
|
#endif // _OPENCV_BLENDERS_HPP_
|