From 8d6db841c6c586084e556c68d83d0048b4c12977 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Tue, 21 Apr 2020 12:39:49 +0200
Subject: [PATCH] [engine.predictor] Avoid unnecessary squeeze() and fix
crashing test on CI
---
bob/ip/binseg/engine/predictor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/bob/ip/binseg/engine/predictor.py b/bob/ip/binseg/engine/predictor.py
index 33a91d0b..804352eb 100644
--- a/bob/ip/binseg/engine/predictor.py
+++ b/bob/ip/binseg/engine/predictor.py
@@ -100,7 +100,7 @@ def _save_overlayed_png(stem, image, prob, output_folder):
"""
image = VF.to_pil_image(image)
- prob = VF.to_pil_image(prob.cpu().squeeze(0))
+ prob = VF.to_pil_image(prob.cpu())
# color and overlay
prob_green = PIL.ImageOps.colorize(prob, (0, 0, 0), (0, 255, 0))
--
GitLab