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)
还没有任何评论哟~
