Advertisement

tensorflow实践-ResNet50预训练权重在图像分类中的应用

阅读量:
复制代码
    import tensorflow as tf
    from tensorflow.python.training import moving_averages
    
    fc_initializer = tf.contrib.layers.xavier_initializer
    conv2d_initializer = tf.contrib.layers.xavier_initializer_conv2d
    
    # create weight variable
    def create_var(name, shape, initializer, trainable=True):
    return tf.get_variable(name, shape=shape, dtype=tf.float32,
                           initializer=initializer, trainable=trainable)
    
    # conv2d layer
    def conv2d(x, num_outputs, kernel_size, stride=1, scope="conv2d"):
    num_inputs = x.get_shape()[-1]
    with tf.va

全部评论 (0)

还没有任何评论哟~