예제 #1
0
 def testEndToEnd3D(self):
     images = tf.random_uniform([1, 299, 299, 299, 3])
     hparams = revnet.revnet_base()
     hparams.dim = '3d'
     hparams.mode = tf.estimator.ModeKeys.TRAIN
     logits = revnet.revnet(images, hparams)
     self.assertEquals(logits.shape, [1, 1, 1, 1, 3328])
예제 #2
0
 def testEndToEnd3D(self):
   images = tf.random_uniform([1, 299, 299, 299, 3])
   hparams = revnet.revnet_base()
   hparams.dim = '3d'
   hparams.mode = tf.estimator.ModeKeys.TRAIN
   logits = revnet.revnet(images, hparams)
   self.assertEquals(logits.shape, [1, 1, 1, 1, 3328])