|
import numpy as np
|
|
|
|
|
|
def pil_img_to_numpy(pil_img):
|
|
"""convert a PIL image to numpy nd-array
|
|
|
|
:param pil_img: a PIL image
|
|
:type pil_img: PIL.Image
|
|
:return: a nd-array
|
|
:rtype: numpy.ndarray
|
|
"""
|
|
np_img = np.array(pil_img)
|
|
np_img = np.rollaxis(np_img, 2) # HWC to CHW
|
|
return np_img
|