如何将图像转化为4D的numpy数组

Question

目前在进行图像分类任务，基于Keras框架，网络中原始代码使用的公开数据集，我希望能用本地自己的数据集进行替换，然后首要的目的就是把图片转换成4D张量，不知道有没有大佬支支招

Answer 1

首先，确定你说的4D应该是4维的意思。

普通的彩色图片一般是3维的，需要增加一维可以使用reshape函数，假设原图的numpy格式图像为image，那么具体操作如下：

image_out = tf.reshape(image, [-1, w, h, 3])