def testAddInputDistortions(self): with tf.Graph().as_default(): with tf.Session() as sess: retrain.add_input_distortions(True, 10, 10, 10) self.assertIsNotNone( sess.graph.get_tensor_by_name('DistortJPGInput:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('DistortResult:0'))
def testAddInputDistortions(self): with tf.Graph().as_default(): with tf.Session() as sess: retrain.add_input_distortions(True, 10, 10, 10) self.assertIsNotNone(sess.graph.get_tensor_by_name('DistortJPGInput:0')) self.assertIsNotNone(sess.graph.get_tensor_by_name('DistortResult:0'))
def testAddInputDistortions(self): with tf.Graph().as_default(): with tf.Session() as sess: retrain.add_input_distortions(True, 10, 10, 10, "jpeg", "distort") self.assertIsNotNone(sess.graph.get_tensor_by_name("jpeg:0")) self.assertIsNotNone(sess.graph.get_tensor_by_name("distort:0"))