added reference.png diagnostic output

In addition to training inputs, also now outputs an image showing the ground truth.
atriumlts · Oct 8, 2016 · 0852d4b · 0852d4b
1 parent b94c9af
commit 0852d4b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/model.py b/model.py
@@ -105,6 +105,8 @@ def train(self, config):
         sample_images = np.array(sample).astype(np.float32)
         sample_input_images = np.array(sample_inputs).astype(np.float32)
 
+        save_images(sample_images, [8, 8], './samples/reference.png')
+
         counter = 1
         start_time = time.time()