Spaces:
Running
on
A10G
Running
on
A10G
| class MaskedImage { | |
| public: | |
| MaskedImage() : m_image(), m_mask(), m_global_mask(), m_image_grady(), m_image_gradx(), m_image_grad_computed(false) { | |
| // pass | |
| } | |
| MaskedImage(cv::Mat image, cv::Mat mask) : m_image(image), m_mask(mask), m_image_grad_computed(false) { | |
| // pass | |
| } | |
| MaskedImage(cv::Mat image, cv::Mat mask, cv::Mat global_mask) : m_image(image), m_mask(mask), m_global_mask(global_mask), m_image_grad_computed(false) { | |
| // pass | |
| } | |
| MaskedImage(cv::Mat image, cv::Mat mask, cv::Mat global_mask, cv::Mat grady, cv::Mat gradx, bool grad_computed) : | |
| m_image(image), m_mask(mask), m_global_mask(global_mask), | |
| m_image_grady(grady), m_image_gradx(gradx), m_image_grad_computed(grad_computed) { | |
| // pass | |
| } | |
| MaskedImage(int width, int height) : m_global_mask(), m_image_grady(), m_image_gradx() { | |
| m_image = cv::Mat(cv::Size(width, height), CV_8UC3); | |
| m_image = cv::Scalar::all(0); | |
| m_mask = cv::Mat(cv::Size(width, height), CV_8U); | |
| m_mask = cv::Scalar::all(0); | |
| } | |
| inline MaskedImage clone() { | |
| return MaskedImage( | |
| m_image.clone(), m_mask.clone(), m_global_mask.clone(), | |
| m_image_grady.clone(), m_image_gradx.clone(), m_image_grad_computed | |
| ); | |
| } | |
| inline cv::Size size() const { | |
| return m_image.size(); | |
| } | |
| inline const cv::Mat &image() const { | |
| return m_image; | |
| } | |
| inline const cv::Mat &mask() const { | |
| return m_mask; | |
| } | |
| inline const cv::Mat &global_mask() const { | |
| return m_global_mask; | |
| } | |
| inline const cv::Mat &grady() const { | |
| assert(m_image_grad_computed); | |
| return m_image_grady; | |
| } | |
| inline const cv::Mat &gradx() const { | |
| assert(m_image_grad_computed); | |
| return m_image_gradx; | |
| } | |
| inline void init_global_mask_mat() { | |
| m_global_mask = cv::Mat(m_mask.size(), CV_8U); | |
| m_global_mask.setTo(cv::Scalar(0)); | |
| } | |
| inline void set_global_mask_mat(const cv::Mat &other) { | |
| m_global_mask = other; | |
| } | |
| inline bool is_masked(int y, int x) const { | |
| return static_cast<bool>(m_mask.at<unsigned char>(y, x)); | |
| } | |
| inline bool is_globally_masked(int y, int x) const { | |
| return !m_global_mask.empty() && static_cast<bool>(m_global_mask.at<unsigned char>(y, x)); | |
| } | |
| inline void set_mask(int y, int x, bool value) { | |
| m_mask.at<unsigned char>(y, x) = static_cast<unsigned char>(value); | |
| } | |
| inline void set_global_mask(int y, int x, bool value) { | |
| m_global_mask.at<unsigned char>(y, x) = static_cast<unsigned char>(value); | |
| } | |
| inline void clear_mask() { | |
| m_mask.setTo(cv::Scalar(0)); | |
| } | |
| inline const unsigned char *get_image(int y, int x) const { | |
| return m_image.ptr<unsigned char>(y, x); | |
| } | |
| inline unsigned char *get_mutable_image(int y, int x) { | |
| return m_image.ptr<unsigned char>(y, x); | |
| } | |
| inline unsigned char get_image(int y, int x, int c) const { | |
| return m_image.ptr<unsigned char>(y, x)[c]; | |
| } | |
| inline int get_image_int(int y, int x, int c) const { | |
| return static_cast<int>(m_image.ptr<unsigned char>(y, x)[c]); | |
| } | |
| bool contains_mask(int y, int x, int patch_size) const; | |
| MaskedImage downsample() const; | |
| MaskedImage upsample(int new_w, int new_h) const; | |
| MaskedImage upsample(int new_w, int new_h, const cv::Mat &new_global_mask) const; | |
| void compute_image_gradients(); | |
| void compute_image_gradients() const; | |
| static const cv::Size kDownsampleKernelSize; | |
| static const int kDownsampleKernel[6]; | |
| private: | |
| cv::Mat m_image; | |
| cv::Mat m_mask; | |
| cv::Mat m_global_mask; | |
| cv::Mat m_image_grady; | |
| cv::Mat m_image_gradx; | |
| bool m_image_grad_computed = false; | |
| }; | |