def _stem(inputs, endpoints, num_filters=64):
with tf.variable_scope('Stem'):
net = layers.conv2d(inputs, num_filters, [7, 7], stride=2, scope='Conv1_7x7')
net = layers.max_pool2d(net, [3, 3], stride=2, scope='Pool1_3x3')
endpoints['Stem'] = net
print("Stem output size: ", net.get_shape())
return net
#@add_arg_scope
评论列表
文章目录