diff options
Diffstat (limited to 'modules/ocv/cv_image_deblurring.h')
-rw-r--r-- | modules/ocv/cv_image_deblurring.h | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/modules/ocv/cv_image_deblurring.h b/modules/ocv/cv_image_deblurring.h new file mode 100644 index 0000000..cd28786 --- /dev/null +++ b/modules/ocv/cv_image_deblurring.h @@ -0,0 +1,67 @@ +/* + * cv_image_deblurring.h - iterative blind deblurring + * + * Copyright (c) 2016-2017 Intel Corporation + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + * Author: Andrey Parfenov <a1994ndrey@gmail.com> + * Author: Wind Yuan <feng.yuan@intel.com> + */ + +#ifndef XCAM_CV_FEATURE_DEBLURRING_H +#define XCAM_CV_FEATURE_DEBLURRING_H + +#include <video_buffer.h> +#include "cv_std.h" +#include "cv_image_process_helper.h" +#include "cv_image_sharp.h" +#include "cv_edgetaper.h" +#include "cv_wiener_filter.h" + +namespace XCam { + +struct CVIDConfig { + int iterations; // number of iterations for IBD algorithm + + CVIDConfig (unsigned int _iterations = 50) + { + iterations = _iterations; + } +}; + +class CVImageDeblurring +{ +public: + explicit CVImageDeblurring (); + void set_config (CVIDConfig config); + CVIDConfig get_config (); + void blind_deblurring (const cv::Mat &blurred, cv::Mat &deblurred, cv::Mat &kernel, int kernel_size = -1, float noise_power = -1.0f, bool use_edgetaper = true); + +private: + void blind_deblurring_one_channel (const cv::Mat &blurred, cv::Mat &kernel, int kernel_size, float noise_power); + int estimate_kernel_size (const cv::Mat &blurred); + void crop_border (cv::Mat &image); + + XCAM_DEAD_COPY (CVImageDeblurring); + + CVIDConfig _config; + SmartPtr<CVImageProcessHelper> _helper; + SmartPtr<CVImageSharp> _sharp; + SmartPtr<CVEdgetaper> _edgetaper; + SmartPtr<CVWienerFilter> _wiener; +}; + +} + +#endif // XCAM_CV_IMAGE_DEBLURRING_H |