Removed print for hist

PrimozGodec · Aug 2, 2017 · 4c0bf8d · 4c0bf8d
1 parent eb9f3c6
commit 4c0bf8d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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