如何根据类别以不同颜色绘制/显示分割任务的掩码

问题描述

我正在做一个语义分割任务,我有 5 个类。但原始图像(卫星图像)已被切割成 15000 个补丁。现在我想将我的蒙版图像绘制或打印成 5 种不同的颜色,因为我有 5 个类。掩码图像的值为 1、2、3、4、5。如果我打印随机图像,值是这样的, [0. 0. 0. 0. 1.].

如何打印 0 的颜色和 1 的颜色以及其他情况?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)