def testHStride(self):
rev_block_input = tf.random_uniform([2, 299, 299, 256])
rev_block_output = revnet.downsample_bottleneck(rev_block_input,
512,
stride=2,
scope='HStride')
self.assertEquals(rev_block_output.get_shape().as_list(),
[2, 150, 150, 512])
def testHStride(self):
rev_block_input = tf.random_uniform([2, 299, 299, 256])
rev_block_output = revnet.downsample_bottleneck(
rev_block_input, 512, stride=2, scope='HStride')
self.assertEquals(rev_block_output.get_shape().as_list(),
[2, 150, 150, 512])
def testH(self):
rev_block_input = tf.random_uniform([1, 299, 299, 3])
rev_block_output = revnet.downsample_bottleneck(rev_block_input, 256)
self.assertEquals(rev_block_output.get_shape().as_list(),
[1, 299, 299, 256])
def testH(self):
rev_block_input = tf.random_uniform([1, 299, 299, 3])
rev_block_output = revnet.downsample_bottleneck(rev_block_input, 256)
self.assertEquals(rev_block_output.get_shape().as_list(),
[1, 299, 299, 256])