python - 通过排除零和 NaN 来调整图像大小

Question

我想调整图像大小，但我的图像有很多零和 NaN。我想通过排除零和 NaN 来调整大小。

到目前为止我拥有的代码是:

# Import Libraries
import numpy as np
import cv2

# Generate sample data with zero and NaNs
a = np.zeros(shape=(8,8))
a[0] = [1,1,3,3,5,5,7,0]
a[1] = [1,0,3,0,5,0,0,7]
a[2] = [1,2,3,4,5,6,7,8]
a[3] = [1,2,3,0,5,6,7,8]
a[4] = [1,2,3,3,5,6,7,8]
a[5] = [1,2,3,4,5,6,7,8]
a[6] = [2,2,3,4,5,0,7,8]
a[7] = [0,2,3,4,5,6,7,8]
a[1:3,5] = np.nan

# Replace nan with zero (or zero with nan)
# a[np.isnan(a)] = 0
a[a==0] = np.nan

# Resize (8x8) array into (4x4)
c = cv2.resize(a,(4,4),cv2.INTER_CUBIC)

问题在于求平均值时包含零和 NaN。我还用 NaN 替换了零，反之亦然。我还尝试了INTER_NEAREST、INTER_LINEAR等，但结果仍然相同。

Answer 1

您可以遵循以下顺序:

1. 用你的 nan 和 0 生成修复蒙版。
2. 申请cv2.Inpaint用插值填充 mask 像素
3. 调整大小