InvalidArgumentError：输入必须为4维[8,6171,4]

Question

我正在运行一个2D卷积网络。我的输入有3维，但是却出现4维错误： dimension error

您可以看到我的输入具有正确的尺寸：

correct input dimension

这是我的代码：

  ngOnInit() {
   this.paramsSubscriptios = this.route.params.subscribe(params => {
      if ((params['country'] !== undefined) && (params['state'] !== null)) {
        this.country = params['country'];
        this.state = params['state'];
      }
    });
  
  }

为什么当我仅构建3-D输入层时要求我提供4-D？

请帮助。 谢谢。

Answer 1

您需要使用

new_image = tf.expand_dims(image,0)

因为模型需要一个数据集而不是单个图像。

Answer 2

通过传递axis=0添加一个外部批处理轴，0是新dim的位置(1, 416, 416, 3) # if -1 will be at end

image_data = tf.expand_dims(image_data, axis=0)