From 44aaa39ccf4b2491d49fd281b6ce0ded881810bd Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 14 Jul 2019 19:04:01 +0300
Subject: [PATCH] Fix Tensorboard RGB channel order

---
 examples/tensorboard_toy.py                  | 11 +++++++----
 trains/binding/frameworks/tensorflow_bind.py |  2 +-
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/examples/tensorboard_toy.py b/examples/tensorboard_toy.py
index 5f044a43..210208c4 100644
--- a/examples/tensorboard_toy.py
+++ b/examples/tensorboard_toy.py
@@ -49,12 +49,15 @@ all_combined = tf.concat(all_distributions, 0)
 tf.summary.histogram("all_combined", all_combined)
 
 # convert to 4d [batch, col, row, RGB-channels]
-image = cv2.imread('./samples/picasso.jpg')
-image = image[:, :, 0][np.newaxis, :, :, np.newaxis]
-# image = image[np.newaxis, :, :, :]  # test greyscale image
+image = cv2.imread('./samples/picasso.jpg')[:, :, ::-1]
+image_gray = image[:, :, 0][np.newaxis, :, :, np.newaxis]
+image_rgba = np.concatenate((image, 255*np.atleast_3d(np.ones(shape=image.shape[:2], dtype=np.uint8))), axis=2)
+image_rgba = image_rgba[np.newaxis, :, :, :]
+image = image[np.newaxis, :, :, :]
 
-# un-comment to add image reporting
 tf.summary.image("test", image, max_outputs=10)
+tf.summary.image("test_gray", image_gray, max_outputs=10)
+tf.summary.image("test_rgba", image_rgba, max_outputs=10)
 
 # Setup a session and summary writer
 summaries = tf.summary.merge_all()
diff --git a/trains/binding/frameworks/tensorflow_bind.py b/trains/binding/frameworks/tensorflow_bind.py
index 2362b81a..1f8697cd 100644
--- a/trains/binding/frameworks/tensorflow_bind.py
+++ b/trains/binding/frameworks/tensorflow_bind.py
@@ -93,7 +93,7 @@ class EventTrainsWriter(object):
         self.histogram_granularity = histogram_granularity
         self.histogram_update_freq_multiplier = histogram_update_freq_multiplier
         self._logger = logger
-        self._visualization_mode = 'BGR'
+        self._visualization_mode = 'RGB'  # 'BGR'
         self._variants = defaultdict(lambda: ())
         self._scalar_report_cache = {}
         self._hist_report_cache = {}