python - python中的圆形中值滤波器

Question

我想创建一个具有给定半径的圆形中值滤波器，而不是来自数组的方形滤波器。这是我迄今为止的尝试:

#   Apply median filter to each image
import matplotlib.pyplot as plt
radius = 25
disk_filter = plt.fspecial('disk', radius)
w1_median_disk = plt.imfilter(w1data, disk_filter, 'replicate')

w2_median_disk = plt.imfilter(w2data, disk_filter, 'replicate')

w1data 和 w2data 是我尝试应用过滤器的二维 numpy 数组。 fspecial 模块来自 Matlab，但我想在我的 Python 代码中使用它(或等效的东西)。有什么想法吗？

我收到错误消息“

disk_filter = plt.fspecial('disk', radius)
AttributeError: 'module' object has no attribute 'fspecial'"

我想知道是否有一个我可以导入的包含 fspecial 的模块，或者 Python 中的等效命令。

Answer 1

如果您愿意安装/使用额外的软件包，我强烈推荐令人惊叹的 skimage适用于 Python 中的任何类型的图像处理!使用类似磁盘的过滤器进行过滤只需两行代码:

import skimage
import skimage.data
import skimage.morphology
import skimage.filters

# load example image
original = skimage.data.camera()

# create disk-like filter footprint with given radius
radius = 10
circle = skimage.morphology.disk(radius)

# apply median filter with given footprint = structuring element = selem
filtered = skimage.filters.median(original, selem = circle)