【TensorFlow报错】ValueError: Dimensions must be equal, but are 3 and 2 for 'random_crop/GreaterEqual'

报错提示:

cifar10_input.py : ValueError: Dimensions must be equal, but are 3 and 2 for 'random_crop/GreaterEqual' (op: 'GreaterEqual') with input shapes: [3], [2].

报错原因:数据应该传入图片的高度、宽度和通道数,但是只传入了高度和宽度导致报错。

解决方案:执行如下修改

# 报错代码
distorted_image = tf.image.random_crop(reshaped_image, [height, width])

# 改正代码
distorted_image = tf.image.random_crop(reshaped_image, [height, width, 3])

 

  • 2
    点赞
  • 1
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

打赏
文章很值,打赏犒劳作者一下
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页

打赏

长行

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者