From 4c0bf8d448e363e5eb3dd93030c8d127dcd0e7f4 Mon Sep 17 00:00:00 2001
From: primozgodec <p.godec9@gmail.com>
Date: Wed, 2 Aug 2017 10:08:44 +0200
Subject: [PATCH] Removed print for hist

---
 src/image_colorization/test.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/image_colorization/test.py b/src/image_colorization/test.py
index eca8a78..888c2b0 100644
--- a/src/image_colorization/test.py
+++ b/src/image_colorization/test.py
@@ -183,7 +183,6 @@ def color_images_part(model):
 
         # for histograms -> transformation from hist to ab
         if model.name == "class_wo_weights" or model.name == "class_with_weights":
-            print("HIST")
             indices = np.argmax(predictions_ab[:, :, :, :], axis=3)
 
             predictions_a = indices // 20 * 10 - 100 + 5