如何从numpy数组中的概率图中获取二进制图像？

Question

嗨我有一个对象的热图（概率图）。 如何将此热图转换为二进制图像？我想取对象的最大区域来增加与右侧图像的交集。 我非常感谢任何帮助。

Answer 1

要二值化，请与阈值进行比较：

thresh = 0.5
bin_map = prob_map > thresh

从那里你可以使用各种各样的解决方案：

Saving a numpy array as an image

我推荐这个来自migas：

from PIL import Image
im = Image.fromarray(A)
im.save("your_file.jpeg")