Skip to content

Commit 4c0bf8d

Browse files
committed
Removed print for hist
1 parent eb9f3c6 commit 4c0bf8d

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

src/image_colorization/test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,6 @@ def color_images_part(model):
183183

184184
# for histograms -> transformation from hist to ab
185185
if model.name == "class_wo_weights" or model.name == "class_with_weights":
186-
print("HIST")
187186
indices = np.argmax(predictions_ab[:, :, :, :], axis=3)
188187

189188
predictions_a = indices // 20 * 10 - 100 + 5

0 commit comments

Comments
 (0)