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

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

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

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

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