keras图像到数组返回维度

时间:2018-06-08 15:16:18

标签: tensorflow machine-learning keras computer-vision

对于尺寸为224 x 224的彩色图像,image.img_to_array如何返回(224,224,3)而不是(3,224,224)?

我期待轴0和轴1对于单一颜色的强度将是2d阵列。我相信我得到的是一个2d数组,每行包含一行颜色。

为什么这一切都混淆了?

1 个答案:

答案 0 :(得分:0)

我相信您可以通过添加参数data_format =“ channels_first”来解决此问题,因此 img_to_array(img,data_format =“ channels_first)

https://rdrr.io/cran/kerasR/man/img_to_array.html