build_resnet.py 文件源码

python
阅读 17 收藏 0 点赞 0 评论 0

项目:tensorflow-litterbox 作者: rwightman 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号