使用 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。