Convolve2d just by using Numpy

by

You could generate the subarrays using as_strided:

import numpy as np

a = np.array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

sub_shape = (3,3)
view_shape = tuple(np.subtract(a.shape, sub_shape) + 1) + sub_shape
strides = a.strides + a.strides

sub_matrices = np.lib.stride_tricks.as_strided(a,view_shape,strides)

To get rid of your second “ugly” sum, alter your einsum so that the output array only has j and k. This implies your second summation.

conv_filter = np.array([[0,-1,0],[-1,5,-1],[0,-1,0]])
m = np.einsum('ij,ijkl->kl',conv_filter,sub_matrices)

# [[ 6  7  8]
#  [11 12 13]
#  [16 17 18]]

More Related Contents:

  1. 2d convolution using python and numpy
  2. Efficiently Using Multiple Numpy Slices for Random Image Cropping
  3. Very large matrices using Python and NumPy
  4. numpy matrix vector multiplication [duplicate]
  5. Numpy matrix to array
  6. how does multiplication differ for NumPy Matrix vs Array classes?
  7. Bin elements per row – Vectorized 2D Bincount for NumPy
  8. Implement MATLAB’s im2col ‘sliding’ in Python
  9. How to find the average colour of an image in Python with OpenCV?
  10. Numpy ‘smart’ symmetric matrix
  11. numpy get index where value is true
  12. NumPy/OpenCV 2: how do I crop non-rectangular region?
  13. Efficiently Calculating a Euclidean Distance Matrix Using Numpy
  14. Block tridiagonal matrix python
  15. how to perform max/mean pooling on a 2d array using numpy
  16. Most efficient way to calculate radial profile
  17. How to get element-wise matrix multiplication (Hadamard product) in numpy?
  18. How can I efficiently process a numpy array in blocks similar to Matlab’s blkproc (blockproc) function
  19. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  20. NumPy array/matrix of mixed types
  21. Combine 3 separate numpy arrays to an RGB image in Python
  22. How to convert a column or row matrix to a diagonal matrix in Python?
  23. numpy with python: convert 3d array to 2d
  24. Convert row vector to column vector in NumPy
  25. Automatic White Balancing with Grayworld assumption
  26. Python, OpenCV: Increasing image brightness without overflowing UINT8 array
  27. Understanding NumPy’s Convolve
  28. Determine if an image exists within a larger image, and if so, find it, using Python
  29. A numpy array unexpectedly changes when changing another one despite being separate
  30. Avoid overflow when adding numpy arrays

Leave a Comment