美文网首页
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