BUG Remove merge artefact.

This commit is contained in:
Stefan van der Walt
2012-06-24 18:03:02 -07:00
parent 00922099d6
commit 87739ed031
-41
View File
@@ -230,44 +230,3 @@ def view_as_windows(arr_in, window_shape):
arr_out = as_strided(arr_in, shape=new_shape, strides=new_strides)
return arr_out
=======
import numpy as np
from numpy.lib import stride_tricks
__all__ = ['view_as_windows']
def view_as_windows(X, win_size=7):
"""Re-stride an array to simulate a sliding window.
Parameters
----------
X : 2D-ndarray
Input image.
win_size : int
Size of the sliding window.
Returns
-------
window : (N, M, win_size, win_size) ndarray
A view on the original data, representing sliding windows. Note:
modifying this view will also modify the original data.
"""
if not X.ndim == 2:
raise ValueError('Input images must be 2-dimensional.')
X = np.ascontiguousarray(X)
r, c = X.shape
strides = X.strides
row_jump, el_jump = strides
half_width = (win_size // 2)
new_strides = (row_jump, el_jump, row_jump, el_jump)
new_rows = r - win_size + 1
new_cols = c - win_size + 1
new_shape = (new_rows, new_cols, win_size, win_size)
windows = stride_tricks.as_strided(X, shape=new_shape, strides=new_strides)
return windows