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