def remove_black_borders(image): y_nonzero, x_nonzero, _ = np.nonzero(image) return image[np.min(y_nonzero):np.max(y_nonzero), np.min(x_nonzero):np.max(x_nonzero)]