美文网首页
tf.nn.conv2d注意事项

tf.nn.conv2d注意事项

作者: yalesaleng | 来源:发表于2018-07-16 16:24 被阅读7次

    例如:

    Output = tf.nn.conv2d(input, weight, strides = [1,1,1,1], padding = ‘SAME’]
    

    一定要注意input是一个4-D数据,
    所以如果你输入的是一副3-D(例如 4806403)图像,
    那么要用tf.reshape来将input扩展到4-D。

    具体code:

    Input = tf.reshape(input, shape = [1,480,640,3])
    

    相关文章

      网友评论

          本文标题:tf.nn.conv2d注意事项

          本文链接:https://www.haomeiwen.com/subject/yxpbpftx.html