def normalize_image(image, # type: np.ndarray depth # type: int ): # type: (...) -> np.ndarray if depth == 1: image = np.packbits(image.flatten()) return image