コード例 #1
0
 def testInit3D(self):
     images = tf.random_uniform([1, 299, 299, 299, 3])
     x1, x2 = revnet.init(images, 32, dim='3d', scope='init3d')
     self.assertEquals(x1.get_shape().as_list(), [1, 74, 74, 74, 16])
     self.assertEquals(x2.get_shape().as_list(), [1, 74, 74, 74, 16])
コード例 #2
0
ファイル: revnet_test.py プロジェクト: chqiwang/tensor2tensor
 def testInit3D(self):
   images = tf.random_uniform([1, 299, 299, 299, 3])
   x1, x2 = revnet.init(images, 32, dim='3d', scope='init3d')
   self.assertEquals(x1.get_shape().as_list(), [1, 74, 74, 74, 16])
   self.assertEquals(x2.get_shape().as_list(), [1, 74, 74, 74, 16])
コード例 #3
0
ファイル: revnet_test.py プロジェクト: yzx1992/tensor2tensor
 def testInit(self):
     images = tf.random_uniform([1, 299, 299, 3])
     x1, x2 = revnet.init(images, 32)
     self.assertEquals(x1.get_shape(), [1, 74, 74, 16])
     self.assertEquals(x2.get_shape(), [1, 74, 74, 16])
コード例 #4
0
 def testInit(self):
   images = tf.random_uniform([1, 299, 299, 3])
   x1, x2 = revnet.init(images, 32)
   self.assertEquals(x1.get_shape(), [1, 74, 74, 16])
   self.assertEquals(x2.get_shape(), [1, 74, 74, 16])