From f73de37ce3ee3bd01afdf9743f398d8c1a03c0c3 Mon Sep 17 00:00:00 2001 From: cclauss Date: Tue, 1 Jan 2019 07:10:54 +0100 Subject: [PATCH] Update image_utils.py --- utils/image_utils.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/utils/image_utils.py b/utils/image_utils.py index 91435a7..c8d4608 100644 --- a/utils/image_utils.py +++ b/utils/image_utils.py @@ -287,12 +287,9 @@ class TFLabConverter(): self.lab_output_tensor = tf_rgb_to_lab()(self.bgr_input_tensor) self.bgr_output_tensor = tf_lab_to_rgb()(self.lab_input_tensor) - - - def bgr2lab(self, bgr): + + def bgr2lab(self, bgr): return self.tf_sess.run(self.lab_output_tensor, feed_dict={self.bgr_input_tensor: bgr}) - - def lab2bgr(self, lab): - return self.tf_sess.run(self.bgr_output_tensor, feed_dict={self.lab_input_tensor: lab}) - - + + def lab2bgr(self, lab): + return self.tf_sess.run(self.bgr_output_tensor, feed_dict={self.lab_input_tensor: lab})