Проблемы с 2D-интерполяцией в Scipy - PullRequest
4 голосов
/ 10 октября 2010

В моем приложении данные отбираются на искаженной сетке, и я хотел бы повторно сопоставить ее с неискаженной сеткой. Чтобы проверить это, я написал эту программу с примерными искажениями и простой функцией в качестве данных:

from __future__ import division

import numpy as np
import scipy.interpolate as intp
import pylab as plt

# Defining some variables:

quadratic = -3/128
linear = 1/16
pn = np.poly1d([quadratic, linear,0])

pixels_x = 50
pixels_y = 30
frame = np.zeros((pixels_x,pixels_y))

x_width= np.concatenate((np.linspace(8,7.8,57) , np.linspace(7.8,8,pixels_y-57)))

def data(x,y):
    z = y*(np.exp(-(x-5)**2/3) + np.exp(-(x)**2/5) + np.exp(-(x+5)**2))
    return(z)

# Generating grid coordinates

yt = np.arange(380,380+pixels_y*4,4)
xt = np.linspace(-7.8,7.8,pixels_x)
X, Y = np.meshgrid(xt,yt)
Y=Y.T
X=X.T

Y_m = np.zeros((pixels_x,pixels_y))
X_m = np.zeros((pixels_x,pixels_y))

# generating distorted grid coordinates:    

for i in range(pixels_y):
    Y_m[:,i] = Y[:,i] - pn(xt)
    X_m[:,i] = np.linspace(-x_width[i],x_width[i],pixels_x)


# Sample data:
for i in range(pixels_y):
    for j in range(pixels_x):
        frame[j,i] = data(X_m[j,i],Y_m[j,i])


Y_m = Y_m.flatten()
X_m = X_m.flatten()
frame = frame.flatten()
##
Y = Y.flatten()
X = X.flatten()
ipf = intp.interp2d(X_m,Y_m,frame)
interpolated_frame = ipf(xt,yt)

На данный момент у меня есть вопросы:

  1. Код работает, но я получаю следующее предупреждение:

    Предупреждение: больше узлов не может быть добавлено, потому что число коэффициентов B-сплайна уже превышает количество точек данных м. Вероятно, причины: либо с или м слишком маленький. (FP> s) kx, ky = 1,1 nx, ny = 54,31 м = 1500 фп = 0,000006 с = 0,000000

Кроме того, появляются некоторые артефакты интерполяции, и я предполагаю, что они связаны с предупреждением - вы, ребята, знаете, что я делаю неправильно?

  1. Для моих реальных приложений кадры должны быть около 500 * 100, но при этом я получаю MemoryError - Есть ли что-то, что я могу сделать, чтобы помочь этому, кроме разделения кадра на несколько частей?

Спасибо!

Ответы [ 3 ]

1 голос
/ 20 октября 2010

для 2d интерполяции, GridData твердый, местный, быстрый. Взгляните на проблема с 2d-интерполяцией в scipy-non-прямоугольной сетке на SO.

1 голос
/ 17 февраля 2016

Эта проблема, скорее всего, связана с использованием bisplrep и bisplev в interp2d . В документах упоминается, что они используют коэффициент сглаживания s = 0,0 и что bisplrep и bisplev должны использоваться напрямую, если требуется больше контроля над s необходимо. В связанных документах упоминается, что с должно быть найдено между (м-кв (2 * м), м + кв (2 * м)) , где m - количество точек, используемых для построения сплайнов. У меня была похожая проблема, и я обнаружил, что она решена при прямом использовании bisplrep и bisplev , где s является необязательным.

0 голосов
/ 10 октября 2010

Возможно, вы захотите взглянуть на следующий метод interp в базовой карте:

mpl_toolkits.basemap.interp http://matplotlib.sourceforge.net/basemap/doc/html/api/basemap_api.html

, если вам действительно не нужна сплайн-интерполяция.

...