Como dividir um Numpy Array (imagem) em blocos para iterar? [duplicado]
Eu tenho uma imagem do tipo Numpy Array que gostaria de dividir em 9 (3 x 3) blocos iguais nos quais posso iterar. Como posso fazer isso?
Aqui está meu código até agora para gerar o numpy.ndarray, mas não consegui dividi-lo:
A matriz de imagens numpy para dividir é th1
import cv2
import numpy as np
# Only for the threshold display
from matplotlib import pyplot as plt
# The Image to be used
image = 'six.png'
# Finding the average greyscale value
image_bgr = cv2.imread(image, cv2.IMREAD_COLOR)
# Calculate the mean of each channel
channels = cv2.mean(image_bgr)
# Type Float
thresh = channels[0]/2
#print (thresh)
# Displaying the threshold value
img = cv2.imread(image,0)
img = cv2.medianBlur(img,5)
# If below then black else white
ret,th1 = cv2.threshold(img,thresh,255,cv2.THRESH_BINARY)
titles = ['Original Image', 'Global Thresholding']
images = [img, th1, ret]
for i in range(2):
plt.subplot(2,2,i+1),plt.imshow(images[i],'gray')
plt.title(titles[i])
plt.xticks([]),plt.yticks([])
# Shows single image on its' own
plt.imshow(images[1], 'gray')
plt.xticks([]),plt.yticks([])
Aqui está a imagem (six.png):
Respostas
O que eu faria é criar uma lista ou um array bidimensional em que salvaria os "tiles" da imagem inteira. Os tiles seriam copiados usando o código abaixo:
height, width, dim = img.shape
image_tile = img[0:height/3, 0:width/3]
Ele cria um novo ndarray image_tile que contém a parte do array de imagem completo que está dentro das coordenadas. Nesse caso, é o ladrilho superior esquerdo.
Todo o exemplo com um loop for se pareceria com isto:
image_tile = []
for i in range(0,3):
for j in range(0,3):
image_tile.append(img[i * height/3:(i+1) * height/3, j * width/3:(j+1) * width/3])
Espero que isso ajude e boa sorte!
Eu encontrei uma solução adaptada de outro lugar e funciona muito bem!
img = th1
numrows, numcols = 3, 3
height = int(img.shape[0] / numrows)
width = int(img.shape[1] / numcols)
for row in range(numrows):
for col in range(numcols):
y0 = row * height
y1 = y0 + height
x0 = col * width
x1 = x0 + width
individual = (img[y0:y1, x0:x1])
plt.imshow(individual, 'gray')
plt.xticks([]),plt.yticks([])
plt.show()