keras-io
/

semantic-segmentation

Image Segmentation

Model card Files Files and versions Community

merve HF staff commited on Nov 26, 2021

Commit

176687d

•

1 Parent(s): f03a9f0

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -52,7 +52,7 @@ class PreTrainedPipeline():
             with io.BytesIO() as out:
                 img.save(out, format="PNG")
                 png_string = out.getvalue()
-                mask = base64.b64encode(png_string)
             mask_codes[f"mask_{cls}"] = mask

             with io.BytesIO() as out:
                 img.save(out, format="PNG")
                 png_string = out.getvalue()
+                mask = base64.b64encode(png_string).decode("utf-8")
             mask_codes[f"mask_{cls}"] = mask