Keras中 img_to_array()函数详解

最新推荐文章于 2024-08-31 16:15:00 发布

AI剑客

最新推荐文章于 2024-08-31 16:15:00 发布

阅读量6.4k

点赞数 2

分类专栏： AI 图像的处理

本文链接：https://blog.csdn.net/qq_43258953/article/details/101481949

版权

本文详细探讨了Keras库中的img_to_array()函数，指出该函数与nd.array()的区别。img_to_array()确保将图像转换为3D Numpy数组，即使原始图像shape为2D。而nd.array()转换时不调整shape。

摘要由CSDN通过智能技术生成

keras中的img_to_array()函数并不等于nd.array().
主要区别在于，img_to_array()不管是2D shape还是3D shape image，返回都是3D Numpy array. 而nd.array()本身不改变shape。

def img_to_array(img, data_format='channels_last', dtype='float32'):
    """Converts a PIL Image instance to a Numpy array.

    # Arguments
        img: PIL Image instance.
        data_format: Image data format,
            either "channels_first" or "channels_last".
        dtype: Dtype to use for the returned array.

    # Returns
        A 3D Numpy array.

    # Raises
        ValueError: if invalid `img` or `data_format` is passed.
    """
    if data_format not in {'channels_first', 'channels_last'}:
        raise ValueError('Unknown data_format: %s' % data_format)
    # Numpy array x has format (height, width, channel)
    # or (channel, height, width)
    # but original PIL image has format