コード例 #1
0
 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])
コード例 #2
0
ファイル: revnet_test.py プロジェクト: chqiwang/tensor2tensor
 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])
コード例 #3
0
 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])
コード例 #4
0
ファイル: revnet_test.py プロジェクト: chqiwang/tensor2tensor
 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])