大きな画像ファイルから作成されたかなり大きな配列を使用しています。メモリの使用量が多すぎるという問題がnumpy.memmap
あり、標準の代わりに配列を使用してみることにしましたnumpy.array
。を作成しmemmap
、画像ファイルからチャンクでデータを読み込むことができましたが、操作の結果を に読み込む方法がわかりませんmemmap
。
たとえば、画像ファイルはnumpy
バイナリ整数配列として読み込まれます。True
指定されたセル数だけセルの任意の領域をバッファリング (拡張) する関数を作成しました。この関数は、入力配列をBoolean
usingに変換しますarray.astype(bool)
。配列によって作成された新しいBoolean
配列を作成するにはどうすればよいですか?array.astype(bool)
numpy.memmap
また、True
指定されたバッファー距離よりも入力配列の端に近いセルがある場合、関数は行や列を配列の端に追加して、既存のTrue
セルの周囲に完全なバッファーを配置できるようにします。これにより、配列の形状が変更されます。の形を変えることは可能numpy.memmap
ですか?
これが私のコードです:
def getArray(dataset):
'''Dataset is an instance of the GDALDataset class from the
GDAL library for working with geospatial datasets
'''
chunks = readRaster.GetArrayParams(dataset, chunkSize=5000)
datPath = re.sub(r'\.\w+$', '_temp.dat', dataset.GetDescription())
pathExists = path.exists(datPath)
arr = np.memmap(datPath, dtype=int, mode='r+',
shape=(dataset.RasterYSize, dataset.RasterXSize))
if not pathExists:
for chunk in chunks:
xOff, yOff, xWidth, yWidth = chunk
chunkArr = readRaster.GetArray(dataset, *chunk)
arr[yOff:yOff + yWidth, xOff:xOff + xWidth] = chunkArr
return arr
def Buffer(arr, dist, ring=False, full=True):
'''Applies a buffer to any non-zero raster cells'''
arr = arr.astype(bool)
nzY, nzX = np.nonzero(arr)
minY = np.amin(nzY)
maxY = np.amax(nzY)
minX = np.amin(nzX)
maxX = np.amax(nzX)
if minY - dist < 0:
arr = np.vstack((np.zeros((abs(minY - dist), arr.shape[1]), bool),
arr))
if maxY + dist >= arr.shape[0]:
arr = np.vstack((arr,
np.zeros(((maxY + dist - arr.shape[0] + 1), arr.shape[1]), bool)))
if minX - dist < 0:
arr = np.hstack((np.zeros((arr.shape[0], abs(minX - dist)), bool),
arr))
if maxX + dist >= arr.shape[1]:
arr = np.hstack((arr,
np.zeros((arr.shape[0], (maxX + dist - arr.shape[1] + 1)), bool)))
if dist >= 0: buffOp = binary_dilation
else: buffOp = binary_erosion
bufDist = abs(dist) * 2 + 1
k = np.ones((bufDist, bufDist))
bufArr = buffOp(arr, k)
return bufArr.astype(int)