python - 使用 python/numpy 进行任意图像切片

Question

我想沿着任意线绘制 2D 图像的 1D 轮廓。下面的代码加载托管在 github 上的图像数据并绘制它：

import urllib
import numpy as np
import matplotlib.pyplot as plt

url = "https://gist.github.com/andreiberceanu/7141843/raw/0b9d50d3d417b1cbe651560470c098700df5a1fc/image.dat"
f = urllib.urlopen(url)
data = np.loadtxt(f)

plt.imshow(data)

例如，上图中的红线是手工绘制的。我想可以以 a*x + b 的形式对其进行参数化。我也猜测某种插值是必要的，因为该线通过可能不是原始二维数据数组的一部分的点。

Answer 1

你想用scipy.ndimage.map_coordinates. 您需要建立一个 2xn 数组，该数组是采样然后执行的坐标map_coordinates(im, samples)。

我觉得这就是：

def sliceImage(I, a, b, *arg, **kws):
    from scipy import linspace, asarray
    from scipy.ndimage import map_coordinates
    from scipy.linalg import norm
    dst = norm(asarray(b) - a) + 1
    return map_coordinates(I, [linspace(strt, end, dst) 
                               for strt, end in zip(a, b)],
                           *arg, **kws)

编辑：进一步考虑，我认为这更优雅：

def sliceImage(I, a, b, *arg, **kws):
    from scipy import linspace, asarray
    from scipy.ndimage import map_coordinates
    from scipy.linalg import norm
    a = asarray(a)
    b = asarray(b)
    dst = norm(b - a) + 1
    return map_coordinates(I, (a[:,newaxis] * linspace(1, 0, dst) +
                               b[:,newaxis] * linspace(0, 1, dst)),
                           *arg, **kws)

编辑：感谢 tcaswell：将 1 添加到dst.