def _block_a(net, scope='BlockA'):
# 35 x 35 x 384 grid
# default padding = SAME
# default stride = 1
with tf.variable_scope(scope):
with tf.variable_scope('Br1_Pool'):
br1 = layers.avg_pool2d(net, [3, 3], scope='Pool1_3x3')
br1 = layers.conv2d(br1, 96, [1, 1], scope='Conv1_1x1')
with tf.variable_scope('Br2_1x1'):
br2 = layers.conv2d(net, 96, [1, 1], scope='Conv1_1x1')
with tf.variable_scope('Br3_3x3'):
br3 = layers.conv2d(net, 64, [1, 1], scope='Conv1_1x1')
br3 = layers.conv2d(br3, 96, [3, 3], scope='Conv2_3x3')
with tf.variable_scope('Br4_3x3Dbl'):
br4 = layers.conv2d(net, 64, [1, 1], scope='Conv1_1x1')
br4 = layers.conv2d(br4, 96, [3, 3], scope='Conv2_3x3')
br4 = layers.conv2d(br4, 96, [3, 3], scope='Conv3_3x3')
net = tf.concat(3, [br1, br2, br3, br4], name='Concat1')
# 35 x 35 x 384
return net
build_inception_v4.py 文件源码
python
阅读 18
收藏 0
点赞 0
评论 0
评论列表
文章目录