最大值、最小值、中值、均值滤波


0 定义及用途

       均值滤波,即以模板内算术平均数的值代替中心像素值,可以降低高斯噪声。

       中值滤波,即以模板内进行有序排列后位于中间的像素值代替中心像素值,可以降低椒盐噪声。

       最大值滤波,即以模板内进行有序排列后最大像素值代替中心像素值,可以去除图像中的暗斑,使亮斑增大。

最小值滤波,即以模板内进行有序排列后最小像素值代替中心像素值,可以去除图像中的亮斑,使暗斑增大。

1 代码

class Filter{

public:

  Filter(int width, int height) : width_(width), height_(height) {

    dst_img_.create(height, width, CV_8UC1);

  }

  ~Filter() {

  }

  virtual cv::Mat& do_filter(cv::Mat& src_img) = 0;

  const int radius_ = 1;  // 3 * 3 model

  int height_;

  int width_;

  cv::Mat dst_img_;

};

 

class MeanFilter : public Filter{

public:

  MeanFilter(int width, int height) : Filter(width, height) {

  };

  ~MeanFilter() {

  };

  cv::Mat& do_filter(cv::Mat& src_img) {

    int template_size = (2 * radius_ + 1)*(2 * radius_ + 1);

    for (int i = 0; i < src_img.rows; ++i) {

      for (int j = 0; j < src_img.cols; ++j) {

        int roi_sum = 0;

        for (int m = -radius_; m <= radius_; m++) {

          int r_offset = m + i;

          r_offset = (r_offset < 0) ? 0 : (r_offset >= src_img.rows ? src_img.rows : r_offset);

          for (int n = -radius_; n <= radius_; n++) {

            int c_offset = n + j;

            c_offset = (c_offset < 0) ? 0 : (c_offset >= src_img.cols ? src_img.cols : c_offset);

            roi_sum += static_cast<int>(src_img.at<unsigned char>(r_offset, c_offset));

          }

        }

        dst_img_.at<unsigned char>(i, j) = roi_sum / template_size;

      }

    }

    return dst_img_;

  }

};

// 均值滤波,最大值滤波,最小值滤波

class MiddleFilter : public Filter {

public:

  enum {

    MIN = 0,

    MIDDLE = 1,

    MAX = 2

  }Method;

  MiddleFilter(int width, int height, int method = MIDDLE) : Filter(width, height),

    method_(method){

  };

  ~MiddleFilter() {

  };

  cv::Mat& do_filter(cv::Mat& src_img) {

    int template_size = (2 * radius_ + 1)*(2 * radius_ + 1);

    unsigned char *template_value = new unsigned char[template_size];

    for (int i = 0; i < src_img.rows; ++i) {

      for (int j = 0; j < src_img.cols; ++j) {

        int index = 0;

        memset(template_value, 0, template_size);

        for (int m = -radius_; m <= radius_; m++) {

          int r_offset = m + i;

          r_offset = (r_offset < 0) ? 0 : (r_offset >= src_img.rows ? src_img.rows : r_offset);

          for (int n = -radius_; n <= radius_; n++) {

            int c_offset = n + j;

            c_offset = (c_offset < 0) ? 0 : (c_offset >= src_img.cols ? src_img.cols : c_offset);

            template_value[index++] = src_img.at<unsigned char>(r_offset, c_offset);

          }

        }

        quick_sort(template_value, 0, template_size - 1);

        if (method_ == MIN) {

          dst_img_.at<uchar>(i, j) = template_value[0];

        } else if (method_ == MIDDLE) {

          dst_img_.at<uchar>(i, j) = template_value[template_size / 2];

        } else {

          dst_img_.at<uchar>(i, j) = template_value[template_size - 1];

        }

      }

    }

    delete[] template_value;

    return dst_img_;

  }

  int method_;

};

 

在Python中,我们可以使用NumPy库和PyTorch库来处理图像滤波操作。以下是各自库中如何实现这些基本滤波方法: **使用NumPy库:** NumPy主要用于数值计算,对于简单的图像滤波操作,你可以创建一个合适的大小的数组来表示过滤核,然后利用其卷积函数`np.convolve()`来应用滤波。 ```python import numpy as np # 假设我们有图片array img def mean_filter(img, kernel_size=3): kernel = np.ones((kernel_size, kernel_size)) / (kernel_size ** 2) filtered_img = np.convolve(img, kernel, mode='same') return filtered_img # 同理,其他滤波中值最大值最小值)可以用类似的方式实现,只是核不同: # 中值滤波:使用大小为kernel_size的二分查找核 def median_filter(img, kernel_size): pass # 最大值/最小值滤波:使用全为1或-1的核 def max_filter(img, kernel_size): kernel = np.ones((kernel_size, kernel_size)) * -np.inf if min_value else np.ones((kernel_size, kernel_size)) ... ``` **使用PyTorch库:** PyTorch在处理图像时通常用于深度学习,但它也可以用于基本的图像处理。你可以使用`F`模块的卷积函数,并配合自定义的过滤核: ```python import torch import torch.nn.functional as F # 假设img是一个PyTorch张量 def mean_filter_torch(img, kernel_size=3): kernel = torch.ones((kernel_size, kernel_size), dtype=torch.float32) / (kernel_size ** 2) filtered_img = F.conv2d(img, kernel, padding=kernel_size // 2) return filtered_img # 使用F.max_pool2d()等函数替换mean_filter_torch()来进行最大值/最小值滤波 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值