图像墙分割成小块示意代码-2

在学习图像处理中遇到一些有意义的处理,这里是图像墙分割成一个个小块,做此记录.

def split2d(img, cell_size, flatten=True):
	h, w = img.shape[:2]
	sx, sy = cell_size
	cells = [np.hsplit(row, w//sx) for row in np.vsplit(img, h//sy)]
	cells = np.array(cells)
	if flatten:
	    cells = cells.reshape(-1, sy, sx)
	return cells