MCPcopy Create free account
hub / github.com/tensorpack/tensorpack / branch

Method branch

examples/HED/hed.py:113–121  ·  view source on GitHub ↗
(name, l, up)

Source from the content-addressed store, hash-verified

111 edgemap = tf.expand_dims(edgemap, 3, name='edgemap4d')
112
113 def branch(name, l, up):
114 with tf.variable_scope(name):
115 l = Conv2D('convfc', l, 1, kernel_size=1, activation=tf.identity,
116 use_bias=True,
117 kernel_initializer=tf.constant_initializer())
118 while up != 1:
119 l = CaffeBilinearUpSample('upsample{}'.format(up), l, 2)
120 up = up // 2
121 return l
122
123 with argscope(Conv2D, kernel_size=3, activation=tf.nn.relu), \
124 argscope([Conv2D, MaxPooling], data_format='NCHW'):

Callers

nothing calls this directly

Calls 3

Conv2DFunction · 0.85
CaffeBilinearUpSampleFunction · 0.85
formatMethod · 0.80

Tested by

no test coverage detected