mindspore.dataset.vision.get_image_num_channels
- mindspore.dataset.vision.get_image_num_channels(image)[源代码]
获取输入图像通道数。
- 参数:
image (Union[numpy.ndarray, PIL.Image.Image]) - 用于获取通道数的图像。
- 返回:
int,输入图像通道数。
- 异常:
RuntimeError - image 参数的维度小于2。
TypeError - image 参数的类型既不是 np.ndarray,也不是 PIL Image。
样例:
>>> import mindspore.dataset.vision as vision >>> from PIL import Image >>> image = Image.open("/path/to/image_file") >>> num_channels = vision.get_image_num_channels(image)