Skip to content
Snippets Groups Projects
Commit 8d6db841 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[engine.predictor] Avoid unnecessary squeeze() and fix crashing test on CI

parent 971f1254
No related branches found
No related tags found
1 merge request!12Streamlining
Pipeline #39076 failed
Stage: build
Hide whitespace changes
Inline Side-by-side
bob/ip/binseg/engine/predictor.py
+ 1
− 1
View file @ 8d6db841
...@@ -100,7 +100,7 @@ def _save_overlayed_png(stem, image, prob, output_folder): ...@@ -100,7 +100,7 @@ def _save_overlayed_png(stem, image, prob, output_folder):
""" """
image = VF.to_pil_image(image) 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 # color and overlay
prob_green = PIL.ImageOps.colorize(prob, (0, 0, 0), (0, 255, 0)) prob_green = PIL.ImageOps.colorize(prob, (0, 0, 0), (0, 255, 0))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment