ProgJ
2018-04-14 12:11
采纳率: 88.1%
浏览 1.3k

急!关于Tensorflow CNN识别手写数字的问题

数据集是在网上下载的mnist里的,图片好像都是黑白的,读取出来的图片没有 通道数 这个维度,但是tf.layers.conv2d函数的input要求必须是4维(数据量,宽度,高度,通道数),我实际读取的只有三个维度(数据量,宽度,高度),这个该怎么处理啊,麻烦大家帮帮忙

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • ProgJ 2018-04-14 15:02

    感谢各位,在conv2d的input做一个reshape【,1】就可以了

    点赞 打赏 评论

相关推荐 更多相似问题