123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320 |
- #ifndef OPENCV_STITCHING_STITCHER_HPP
- #define OPENCV_STITCHING_STITCHER_HPP
- #include "opencv2/core.hpp"
- #include "opencv2/features2d.hpp"
- #include "opencv2/stitching/warpers.hpp"
- #include "opencv2/stitching/detail/matchers.hpp"
- #include "opencv2/stitching/detail/motion_estimators.hpp"
- #include "opencv2/stitching/detail/exposure_compensate.hpp"
- #include "opencv2/stitching/detail/seam_finders.hpp"
- #include "opencv2/stitching/detail/blenders.hpp"
- #include "opencv2/stitching/detail/camera.hpp"
- #if defined(Status)
- # warning Detected X11 'Status' macro definition, it can cause build conflicts. Please, include this header before any X11 headers.
- #endif
- namespace cv {
- class CV_EXPORTS_W Stitcher
- {
- public:
- enum { ORIG_RESOL = -1 };
- enum Status
- {
- OK = 0,
- ERR_NEED_MORE_IMGS = 1,
- ERR_HOMOGRAPHY_EST_FAIL = 2,
- ERR_CAMERA_PARAMS_ADJUST_FAIL = 3
- };
- enum Mode
- {
-
- PANORAMA = 0,
-
- SCANS = 1,
- };
-
-
- static Stitcher createDefault(bool try_use_gpu = false);
-
- static Ptr<Stitcher> create(Mode mode = PANORAMA, bool try_use_gpu = false);
- CV_WRAP double registrationResol() const { return registr_resol_; }
- CV_WRAP void setRegistrationResol(double resol_mpx) { registr_resol_ = resol_mpx; }
- CV_WRAP double seamEstimationResol() const { return seam_est_resol_; }
- CV_WRAP void setSeamEstimationResol(double resol_mpx) { seam_est_resol_ = resol_mpx; }
- CV_WRAP double compositingResol() const { return compose_resol_; }
- CV_WRAP void setCompositingResol(double resol_mpx) { compose_resol_ = resol_mpx; }
- CV_WRAP double panoConfidenceThresh() const { return conf_thresh_; }
- CV_WRAP void setPanoConfidenceThresh(double conf_thresh) { conf_thresh_ = conf_thresh; }
- CV_WRAP bool waveCorrection() const { return do_wave_correct_; }
- CV_WRAP void setWaveCorrection(bool flag) { do_wave_correct_ = flag; }
- detail::WaveCorrectKind waveCorrectKind() const { return wave_correct_kind_; }
- void setWaveCorrectKind(detail::WaveCorrectKind kind) { wave_correct_kind_ = kind; }
- Ptr<detail::FeaturesFinder> featuresFinder() { return features_finder_; }
- const Ptr<detail::FeaturesFinder> featuresFinder() const { return features_finder_; }
- void setFeaturesFinder(Ptr<detail::FeaturesFinder> features_finder)
- { features_finder_ = features_finder; }
- Ptr<detail::FeaturesMatcher> featuresMatcher() { return features_matcher_; }
- const Ptr<detail::FeaturesMatcher> featuresMatcher() const { return features_matcher_; }
- void setFeaturesMatcher(Ptr<detail::FeaturesMatcher> features_matcher)
- { features_matcher_ = features_matcher; }
- const cv::UMat& matchingMask() const { return matching_mask_; }
- void setMatchingMask(const cv::UMat &mask)
- {
- CV_Assert(mask.type() == CV_8U && mask.cols == mask.rows);
- matching_mask_ = mask.clone();
- }
- Ptr<detail::BundleAdjusterBase> bundleAdjuster() { return bundle_adjuster_; }
- const Ptr<detail::BundleAdjusterBase> bundleAdjuster() const { return bundle_adjuster_; }
- void setBundleAdjuster(Ptr<detail::BundleAdjusterBase> bundle_adjuster)
- { bundle_adjuster_ = bundle_adjuster; }
-
- Ptr<WarperCreator> warper() { return warper_; }
- const Ptr<WarperCreator> warper() const { return warper_; }
- void setWarper(Ptr<WarperCreator> creator) { warper_ = creator; }
- Ptr<detail::ExposureCompensator> exposureCompensator() { return exposure_comp_; }
- const Ptr<detail::ExposureCompensator> exposureCompensator() const { return exposure_comp_; }
- void setExposureCompensator(Ptr<detail::ExposureCompensator> exposure_comp)
- { exposure_comp_ = exposure_comp; }
- Ptr<detail::SeamFinder> seamFinder() { return seam_finder_; }
- const Ptr<detail::SeamFinder> seamFinder() const { return seam_finder_; }
- void setSeamFinder(Ptr<detail::SeamFinder> seam_finder) { seam_finder_ = seam_finder; }
- Ptr<detail::Blender> blender() { return blender_; }
- const Ptr<detail::Blender> blender() const { return blender_; }
- void setBlender(Ptr<detail::Blender> b) { blender_ = b; }
-
- CV_WRAP Status estimateTransform(InputArrayOfArrays images);
-
- Status estimateTransform(InputArrayOfArrays images, const std::vector<std::vector<Rect> > &rois);
-
- CV_WRAP Status composePanorama(OutputArray pano);
-
- Status composePanorama(InputArrayOfArrays images, OutputArray pano);
-
- CV_WRAP Status stitch(InputArrayOfArrays images, OutputArray pano);
-
- Status stitch(InputArrayOfArrays images, const std::vector<std::vector<Rect> > &rois, OutputArray pano);
- std::vector<int> component() const { return indices_; }
- std::vector<detail::CameraParams> cameras() const { return cameras_; }
- CV_WRAP double workScale() const { return work_scale_; }
- private:
-
- Status matchImages();
- Status estimateCameraParams();
- double registr_resol_;
- double seam_est_resol_;
- double compose_resol_;
- double conf_thresh_;
- Ptr<detail::FeaturesFinder> features_finder_;
- Ptr<detail::FeaturesMatcher> features_matcher_;
- cv::UMat matching_mask_;
- Ptr<detail::BundleAdjusterBase> bundle_adjuster_;
-
- bool do_wave_correct_;
- detail::WaveCorrectKind wave_correct_kind_;
- Ptr<WarperCreator> warper_;
- Ptr<detail::ExposureCompensator> exposure_comp_;
- Ptr<detail::SeamFinder> seam_finder_;
- Ptr<detail::Blender> blender_;
- std::vector<cv::UMat> imgs_;
- std::vector<std::vector<cv::Rect> > rois_;
- std::vector<cv::Size> full_img_sizes_;
- std::vector<detail::ImageFeatures> features_;
- std::vector<detail::MatchesInfo> pairwise_matches_;
- std::vector<cv::UMat> seam_est_imgs_;
- std::vector<int> indices_;
- std::vector<detail::CameraParams> cameras_;
- double work_scale_;
- double seam_scale_;
- double seam_work_aspect_;
- double warped_image_scale_;
- };
- CV_EXPORTS_W Ptr<Stitcher> createStitcher(bool try_use_gpu = false);
- }
- #endif
|