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])
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])