rknn-toolkit2-v2.1.0-2024-08-08
/
rknpu2
/examples
/3rdparty
/opencv
/opencv-linux-aarch64
/include
/opencv2
/videostab
/stabilizer.hpp
/*M/////////////////////////////////////////////////////////////////////////////////////// | |
// | |
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. | |
// | |
// By downloading, copying, installing or using the software you agree to this license. | |
// If you do not agree to this license, do not download, install, | |
// copy or use the software. | |
// | |
// | |
// License Agreement | |
// For Open Source Computer Vision Library | |
// | |
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved. | |
// Copyright (C) 2009-2011, Willow Garage Inc., all rights reserved. | |
// Third party copyrights are property of their respective owners. | |
// | |
// Redistribution and use in source and binary forms, with or without modification, | |
// are permitted provided that the following conditions are met: | |
// | |
// * Redistribution's of source code must retain the above copyright notice, | |
// this list of conditions and the following disclaimer. | |
// | |
// * Redistribution's in binary form must reproduce the above copyright notice, | |
// this list of conditions and the following disclaimer in the documentation | |
// and/or other materials provided with the distribution. | |
// | |
// * The name of the copyright holders may not be used to endorse or promote products | |
// derived from this software without specific prior written permission. | |
// | |
// This software is provided by the copyright holders and contributors "as is" and | |
// any express or implied warranties, including, but not limited to, the implied | |
// warranties of merchantability and fitness for a particular purpose are disclaimed. | |
// In no event shall the Intel Corporation or contributors be liable for any direct, | |
// indirect, incidental, special, exemplary, or consequential damages | |
// (including, but not limited to, procurement of substitute goods or services; | |
// loss of use, data, or profits; or business interruption) however caused | |
// and on any theory of liability, whether in contract, strict liability, | |
// or tort (including negligence or otherwise) arising in any way out of | |
// the use of this software, even if advised of the possibility of such damage. | |
// | |
//M*/ | |
namespace cv | |
{ | |
namespace videostab | |
{ | |
//! @addtogroup videostab | |
//! @{ | |
class CV_EXPORTS StabilizerBase | |
{ | |
public: | |
virtual ~StabilizerBase() {} | |
void setLog(Ptr<ILog> ilog) { log_ = ilog; } | |
Ptr<ILog> log() const { return log_; } | |
void setRadius(int val) { radius_ = val; } | |
int radius() const { return radius_; } | |
void setFrameSource(Ptr<IFrameSource> val) { frameSource_ = val; } | |
Ptr<IFrameSource> frameSource() const { return frameSource_; } | |
void setMotionEstimator(Ptr<ImageMotionEstimatorBase> val) { motionEstimator_ = val; } | |
Ptr<ImageMotionEstimatorBase> motionEstimator() const { return motionEstimator_; } | |
void setDeblurer(Ptr<DeblurerBase> val) { deblurer_ = val; } | |
Ptr<DeblurerBase> deblurrer() const { return deblurer_; } | |
void setTrimRatio(float val) { trimRatio_ = val; } | |
float trimRatio() const { return trimRatio_; } | |
void setCorrectionForInclusion(bool val) { doCorrectionForInclusion_ = val; } | |
bool doCorrectionForInclusion() const { return doCorrectionForInclusion_; } | |
void setBorderMode(int val) { borderMode_ = val; } | |
int borderMode() const { return borderMode_; } | |
void setInpainter(Ptr<InpainterBase> val) { inpainter_ = val; } | |
Ptr<InpainterBase> inpainter() const { return inpainter_; } | |
protected: | |
StabilizerBase(); | |
void reset(); | |
Mat nextStabilizedFrame(); | |
bool doOneIteration(); | |
virtual void setUp(const Mat &firstFrame); | |
virtual Mat estimateMotion() = 0; | |
virtual Mat estimateStabilizationMotion() = 0; | |
void stabilizeFrame(); | |
virtual Mat postProcessFrame(const Mat &frame); | |
void logProcessingTime(); | |
Ptr<ILog> log_; | |
Ptr<IFrameSource> frameSource_; | |
Ptr<ImageMotionEstimatorBase> motionEstimator_; | |
Ptr<DeblurerBase> deblurer_; | |
Ptr<InpainterBase> inpainter_; | |
int radius_; | |
float trimRatio_; | |
bool doCorrectionForInclusion_; | |
int borderMode_; | |
Size frameSize_; | |
Mat frameMask_; | |
int curPos_; | |
int curStabilizedPos_; | |
bool doDeblurring_; | |
Mat preProcessedFrame_; | |
bool doInpainting_; | |
Mat inpaintingMask_; | |
Mat finalFrame_; | |
std::vector<Mat> frames_; | |
std::vector<Mat> motions_; // motions_[i] is the motion from i-th to i+1-th frame | |
std::vector<float> blurrinessRates_; | |
std::vector<Mat> stabilizedFrames_; | |
std::vector<Mat> stabilizedMasks_; | |
std::vector<Mat> stabilizationMotions_; | |
clock_t processingStartTime_; | |
}; | |
class CV_EXPORTS OnePassStabilizer : public StabilizerBase, public IFrameSource | |
{ | |
public: | |
OnePassStabilizer(); | |
void setMotionFilter(Ptr<MotionFilterBase> val) { motionFilter_ = val; } | |
Ptr<MotionFilterBase> motionFilter() const { return motionFilter_; } | |
virtual void reset() CV_OVERRIDE; | |
virtual Mat nextFrame() CV_OVERRIDE { return nextStabilizedFrame(); } | |
protected: | |
virtual void setUp(const Mat &firstFrame) CV_OVERRIDE; | |
virtual Mat estimateMotion() CV_OVERRIDE; | |
virtual Mat estimateStabilizationMotion() CV_OVERRIDE; | |
virtual Mat postProcessFrame(const Mat &frame) CV_OVERRIDE; | |
Ptr<MotionFilterBase> motionFilter_; | |
}; | |
class CV_EXPORTS TwoPassStabilizer : public StabilizerBase, public IFrameSource | |
{ | |
public: | |
TwoPassStabilizer(); | |
void setMotionStabilizer(Ptr<IMotionStabilizer> val) { motionStabilizer_ = val; } | |
Ptr<IMotionStabilizer> motionStabilizer() const { return motionStabilizer_; } | |
void setWobbleSuppressor(Ptr<WobbleSuppressorBase> val) { wobbleSuppressor_ = val; } | |
Ptr<WobbleSuppressorBase> wobbleSuppressor() const { return wobbleSuppressor_; } | |
void setEstimateTrimRatio(bool val) { mustEstTrimRatio_ = val; } | |
bool mustEstimateTrimaRatio() const { return mustEstTrimRatio_; } | |
virtual void reset() CV_OVERRIDE; | |
virtual Mat nextFrame() CV_OVERRIDE; | |
protected: | |
void runPrePassIfNecessary(); | |
virtual void setUp(const Mat &firstFrame) CV_OVERRIDE; | |
virtual Mat estimateMotion() CV_OVERRIDE; | |
virtual Mat estimateStabilizationMotion() CV_OVERRIDE; | |
virtual Mat postProcessFrame(const Mat &frame) CV_OVERRIDE; | |
Ptr<IMotionStabilizer> motionStabilizer_; | |
Ptr<WobbleSuppressorBase> wobbleSuppressor_; | |
bool mustEstTrimRatio_; | |
int frameCount_; | |
bool isPrePassDone_; | |
bool doWobbleSuppression_; | |
std::vector<Mat> motions2_; | |
Mat suppressedFrame_; | |
}; | |
//! @} | |
} // namespace videostab | |
} // namespace cv | |