Scipy径向基函数中的Python MemoryError(scipy.interpolate.rbf)

时间:2012-08-08 13:15:57

标签: python memory numpy scipy interpolation

我正在尝试使用Scipy Radial Basis Function(Rbf)插值表示2D表面的不太大(~10,000个样本)的pointcloud。我得到了一些好的结果,但是在我的上一个数据集中,我一直得到MemoryError,即使错误在执行期间几乎立即出现(RAM显然没有被吃掉)。

我决定从Scipy中删除rbf.py文件的副本,首先填写一些非常有用的打印语句。通过逐行分解_euclidean_norm方法,如下所示:

def _euclidean_norm(self, x1, x2):
    d = x1 - x2
    s = d**2
    su = s.sum(axis=0)
    sq = sqrt(su)
    return sq

我在第一行收到错误:

File "C:\MyRBF.py", line 68, in _euclidean_norm
    d = x1 - x2
MemoryError

该规范以[[x1,y1],[x2,y2],[x3,y3],...,[xn,yn]]和X2形式的数组X1调用,即X1在Rbf类中通过以​​下方法进行转置,已经被我用于调试目的:

def _call_norm(self, x1, x2):
    print x1.shape
    print x2.shape
    print

    if len(x1.shape) == 1:
        x1 = x1[newaxis, :]
    if len(x2.shape) == 1:
        x2 = x2[newaxis, :]
    x1 = x1[..., :, newaxis]
    x2 = x2[..., newaxis, :]

    print x1.shape
    print x2.shape
    print

    return self._euclidean_norm(x1, x2)

请注意我打印输入的形状。使用我当前的数据集,这就是我得到的(我手动添加了注释):

(2, 10744)         ## Input array of 10744 x,y pairs
(2, 10744)         ## The same array, which is to be "reshaped/transposed"

(2, 10744, 1)      ## The first "reshaped/transposed" form of the array
(2, 1, 10744)      ## The second "reshaped/transposed" form of the array

根据文献记载,理由是得到“从x1中的每个点到x2中的每个点的距离的矩阵”,这意味着,因为阵列是相同的,所以每对之间的距离矩阵入口数组(包含X和Y维度)。

我用更小的数组(例如形状(2,5,1)和(2,1,5))手动测试了操作,减法工作。

如何找出不使用我的数据集的原因?还有其他明显错误吗?我应该检查我的数据集的某种形式的病态调节,还是对它进行一些预处理?我认为它的条件很好,因为我可以用3D绘制它,而且cloudpoint在视觉上非常好。

非常感谢任何帮助。

感谢阅读。

1 个答案:

答案 0 :(得分:7)

您的数据集应该没问题:出现错误是因为您没有足够的RAM来存储减法的结果。

根据广播规则,结果将具有形状

 (2, 10744,     1)
-(2,     1, 10744)
------------------
 (2, 10744, 10744)

假设这些是dtype float64的数组,则需要2 * 10744 ** 2 * 8 = 1.72 GiB的可用内存。如果没有足够的可用内存,numpy将无法分配输出数组,并且会立即失败并显示您看到的错误。