Как оптимизировать NumPy Среднее окно - PullRequest
Новая
       0

Как оптимизировать NumPy Среднее окно

0 голосов
/ 29 сентября 2018

Я пытаюсь получить новый растр, делающий цикл по всему массиву и использующий окно поиска 25x25 пикселей.Интересно, знаете ли вы лучший способ сделать это, потому что это занимает слишком много времени с моим подходом. 

import sys
import os
import numpy as np
import math

from osgeo import gdal, osr, gdal_array, gdalnumeric
from osgeo.gdalnumeric import *

numpy.warnings.filterwarnings('ignore')

def mean_neighbors(M,x,y,w=1):
    l = []
    for i in range(max(0,x-w), x+(w+1)):
        for j in range(max(0,y-w), y+(w+1)):
            try:
                t = M[i][j]
                l.append(t)
            except IndexError:
                pass
    return np.mean(l)

raster_file = gdal.Open('image.tif', gdal.GA_ReadOnly)
rst         = gdalnumeric.BandReadAsArray(raster_file.GetRasterBand(1))

cob = np.zeros(rst.shape)

for i in range(rst.shape[0]):
    for j in range(rst.shape[1]):
        cob[i][j] = mean_neighbors(rst, i, j, 25) # want to optimize this function

1 Ответ

0 голосов
/ 29 сентября 2018

Перемещение окна означает очень распространенную функцию, которую вам не нужно писать самостоятельно.Вот две быстрые реализации, которые вы можете использовать:

...