def _fully_connected_ST(self, x, out_dim):
"""FullyConnected layer for final output of the localization network in the spatial transformer"""
x = tf.reshape(x, [self.hps.batch_size, -1])
w = tf.get_variable(
'DW2', [x.get_shape()[1], out_dim],
initializer=tf.uniform_unit_scaling_initializer(factor=1.0))
initial = np.array([[1., 0, 0], [0, 1., 0]])
initial = initial.astype('float32')
initial = initial.flatten()
b = tf.get_variable('biases2', [out_dim],
initializer=tf.constant_initializer(initial))
return tf.nn.xw_plus_b(x, w, b)
评论列表
文章目录