From 001eb2b0a4a9d751ae96f322c7b677547b7d8920 Mon Sep 17 00:00:00 2001
From: Daniel Ecer <de-code@users.noreply.github.com>
Date: Tue, 1 Aug 2017 09:03:01 +0100
Subject: [PATCH] include output layer names for RGB mode
---
sciencebeam_gym/trainer/models/pix2pix/pix2pix_model.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/sciencebeam_gym/trainer/models/pix2pix/pix2pix_model.py b/sciencebeam_gym/trainer/models/pix2pix/pix2pix_model.py
index 3e0aed2..c3b0d52 100644
--- a/sciencebeam_gym/trainer/models/pix2pix/pix2pix_model.py
+++ b/sciencebeam_gym/trainer/models/pix2pix/pix2pix_model.py
@@ -382,6 +382,7 @@ class Model(object):
else:
with tf.name_scope('evaluation'):
if self.dimension_colors:
+ tensors.output_layer_labels = tf.constant(self.dimension_labels)
colors_tensor = tf.constant(
self.dimension_colors_with_unknown,
dtype=tf.float32
--
GitLab