123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174 |
- #ifndef __OPENCV_VIDEOSTAB_MOTION_STABILIZING_HPP__
- #define __OPENCV_VIDEOSTAB_MOTION_STABILIZING_HPP__
- #include <vector>
- #include <utility>
- #include "opencv2/core.hpp"
- #include "opencv2/videostab/global_motion.hpp"
- namespace cv
- {
- namespace videostab
- {
- class CV_EXPORTS IMotionStabilizer
- {
- public:
- virtual ~IMotionStabilizer() {}
-
- virtual void stabilize(
- int size, const std::vector<Mat> &motions, std::pair<int,int> range,
- Mat *stabilizationMotions) = 0;
- };
- class CV_EXPORTS MotionStabilizationPipeline : public IMotionStabilizer
- {
- public:
- void pushBack(Ptr<IMotionStabilizer> stabilizer) { stabilizers_.push_back(stabilizer); }
- bool empty() const { return stabilizers_.empty(); }
- virtual void stabilize(
- int size, const std::vector<Mat> &motions, std::pair<int,int> range,
- Mat *stabilizationMotions);
- private:
- std::vector<Ptr<IMotionStabilizer> > stabilizers_;
- };
- class CV_EXPORTS MotionFilterBase : public IMotionStabilizer
- {
- public:
- virtual ~MotionFilterBase() {}
- virtual Mat stabilize(
- int idx, const std::vector<Mat> &motions, std::pair<int,int> range) = 0;
- virtual void stabilize(
- int size, const std::vector<Mat> &motions, std::pair<int,int> range,
- Mat *stabilizationMotions);
- };
- class CV_EXPORTS GaussianMotionFilter : public MotionFilterBase
- {
- public:
- GaussianMotionFilter(int radius = 15, float stdev = -1.f);
- void setParams(int radius, float stdev = -1.f);
- int radius() const { return radius_; }
- float stdev() const { return stdev_; }
- virtual Mat stabilize(
- int idx, const std::vector<Mat> &motions, std::pair<int,int> range);
- private:
- int radius_;
- float stdev_;
- std::vector<float> weight_;
- };
- inline GaussianMotionFilter::GaussianMotionFilter(int _radius, float _stdev) { setParams(_radius, _stdev); }
- class CV_EXPORTS LpMotionStabilizer : public IMotionStabilizer
- {
- public:
- LpMotionStabilizer(MotionModel model = MM_SIMILARITY);
- void setMotionModel(MotionModel val) { model_ = val; }
- MotionModel motionModel() const { return model_; }
- void setFrameSize(Size val) { frameSize_ = val; }
- Size frameSize() const { return frameSize_; }
- void setTrimRatio(float val) { trimRatio_ = val; }
- float trimRatio() const { return trimRatio_; }
- void setWeight1(float val) { w1_ = val; }
- float weight1() const { return w1_; }
- void setWeight2(float val) { w2_ = val; }
- float weight2() const { return w2_; }
- void setWeight3(float val) { w3_ = val; }
- float weight3() const { return w3_; }
- void setWeight4(float val) { w4_ = val; }
- float weight4() const { return w4_; }
- virtual void stabilize(
- int size, const std::vector<Mat> &motions, std::pair<int,int> range,
- Mat *stabilizationMotions);
- private:
- MotionModel model_;
- Size frameSize_;
- float trimRatio_;
- float w1_, w2_, w3_, w4_;
- std::vector<double> obj_, collb_, colub_;
- std::vector<int> rows_, cols_;
- std::vector<double> elems_, rowlb_, rowub_;
- void set(int row, int col, double coef)
- {
- rows_.push_back(row);
- cols_.push_back(col);
- elems_.push_back(coef);
- }
- };
- CV_EXPORTS Mat ensureInclusionConstraint(const Mat &M, Size size, float trimRatio);
- CV_EXPORTS float estimateOptimalTrimRatio(const Mat &M, Size size);
- }
- }
- #endif
|