例如:
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])
例如:
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
网友评论