void FiltMaxImg(const Img& in, const int k_width, const int k_height, const int channel, Img& out) {
const int width = in.width();
const int height = in.height();
const int num_channel = in.channel();
assert(num_channel > channel);
out.Reshape(width, height, 1);
const double* in_data = in.data();
double* out_data = out.mutable_data();
for(int h = 0; h < height; h++) {
for (int w = 0; w < width; w++) {
int w_start = w - ((k_width) / 2);
int h_start = h - ((k_height) / 2);
int kw_start = 0;
int kh_start = 0;
int kw_end = k_width;
int kh_end = k_height;
double max_value = -MAX_DOUBLE;
for (int i = kh_start; i < kh_end; i++ ) {
for (int j = kw_start; j < kw_end; j++) {
int w_temp = w_start + j;
int h_temp = h_start + i;
if (w_temp >=0 && w_temp < width && h_temp >= 0 && h_temp < height) {
max_value = std::max(max_value, in_data[h_temp * width + w_temp]);
}
}
}
out_data[h * width + w] = max_value;
}
}
}
void FiltMedImg(const Img& in, const int k_width, const int k_height, const int channel, Img& out) {
const int width = in.width();
const int height = in.height();
const int num_channel = in.channel();
assert(num_channel > channel);
out.Reshape(width, height, 1);
const double* in_data = in.data();
double* out_data = out.mutable_data();
for(int h = 0; h < height; h++) {
for (int w = 0; w < width; w++) {
int w_start = w - ((k_width) / 2);
int h_start = h - ((k_height) / 2);
int kw_start = 0;
int kh_start = 0;
int kw_end = k_width;
int kh_end = k_height;
std::vector<double> temp_list;
for (int i = kh_start; i < kh_end; i++ ) {
for (int j = kw_start; j < kw_end; j++) {
int w_temp = w_start + j;
int h_temp = h_start + i;
if (w_temp >=0 && w_temp < width && h_temp >= 0 && h_temp < height) {
temp_list.push_back(in_data[h_temp * width + w_temp]);
}
}
}
std::sort(temp_list.begin(), temp_list.end());
out_data[h * width + w] = temp_list[temp_list.size()/2];
}
}
}
void FiltImg(const Img& in, const Img& filter, const int channel, Img& out) {
const int width = in.width();
const int height = in.height();
const int num_channel = in.channel();
assert(num_channel > channel);
out.Reshape(width, height, 1);
const double* in_data = in.data() + channel * out.dim();
const double* filter_data = filter.data();
double* out_data = out.mutable_data();
const int k_width = filter.width();
const int k_height = filter.height();
for(int h = 0; h < height; h++) {
for (int w = 0; w < width; w++) {
int w_start = w - ((k_width) / 2);
int h_start = h - ((k_height) / 2);
int kw_start = 0;
int kh_start = 0;
int kw_end = k_width;
int kh_end = k_height;
double sum = 0;
for (int i = kh_start; i < kh_end; i++ ) {
for (int j = kw_start; j < kw_end; j++) {
int w_temp = w_start + j;
int h_temp = h_start + i;
if (w_temp >=0 && w_temp < width && h_temp >= 0 && h_temp < height) {
sum += in_data[h_temp * width + w_temp] * filter_data[i * k_width + j];
//printf("%f x %f, ", in_data[h_temp * width + w_temp], filter_data[i * k_width + j]);
//printf("%d %d, \n", h_temp, w_temp);
}
}
}
out_data[h * width + w] = sum;
}
}
}
