banxia1995 2019-02-28 20:24 采纳率: 0%
浏览 3632

tensorflow处理图片时的维度不匹配问题

tensorflow处理图片时的维度不匹配问题,如下是报错信息

tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be rank 4 but is rank 3 for 'Conv2D' (op: 'Conv2D') with input shapes: [1,1,3], [5,5,3,32].

源码如下所示:

train_data_node=tf.placeholder(tf.float32,shape=(None,IMG_PATCH_SIZE,IMG_PATCH_SIZE,NUM_CHANNEL))
train_label_node=tf.placeholder(tf.float32,shape=(BATCH_SIZE,NUM_LABEL))
train_all_data_node=tf.constant(train_data)

def extract_data():
imgs=[]
data_list=[]
training_size, img_train_array,img_train_map_array= read_train_from_txt_file(train_txt_filename)
for i in range(0,training_size):
    image_filename = img_train_array[i]
    if os.path.isfile(image_filename):
        print('Loading:'+ image_filename)
        img_file = cv.imread(image_filename)
        img_file=np.array(img_file)
        imgs.append(img_file)
    else:
        print('File' + image_filename + 'does not exist!')
num_img = len(imgs)
for j in range(num_img):
    img_patches = img_crop(imgs[j])
for k in range(len(img_patches)):
    for m in range(len(img_patches[k])):
        data=img_patches[k][m]
        data_list.append(data)
        data_list=np.asarray(data_list)
        data_list=np.float32(data_list)
# return  np.asarray(data)
return data_list

新手一个,求帮助!!!!!
  • 写回答

1条回答 默认 最新

  • qq_28888837 博客专家认证 2019-03-01 16:27
    关注

    你可以把输入的扩展为4维的

    评论

报告相同问题?

悬赏问题

  • ¥15 有兄弟姐妹会用word插图功能制作类似citespace的图片吗?
  • ¥200 uniapp长期运行卡死问题解决
  • ¥15 请教:如何用postman调用本地虚拟机区块链接上的合约?
  • ¥15 为什么使用javacv转封装rtsp为rtmp时出现如下问题:[h264 @ 000000004faf7500]no frame?
  • ¥15 乘性高斯噪声在深度学习网络中的应用
  • ¥15 关于docker部署flink集成hadoop的yarn,请教个问题 flink启动yarn-session.sh连不上hadoop,这个整了好几天一直不行,求帮忙看一下怎么解决
  • ¥15 深度学习根据CNN网络模型,搭建BP模型并训练MNIST数据集
  • ¥15 C++ 头文件/宏冲突问题解决
  • ¥15 用comsol模拟大气湍流通过底部加热(温度不同)的腔体
  • ¥50 安卓adb backup备份子用户应用数据失败