Spaces:
Runtime error
Runtime error
File size: 3,830 Bytes
bb88c4d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
#pragma once
#include <opencv2/core.hpp>
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;
};
|