python显示出labelme产生的label.png以及mask rcnn检测后提取的mask

将mask rcnn中的mask（掩码）提取显示保存为图片
将labelme中label.png的显示保存为图片

我们在进行mask rcnn训练后，产生了带有mask的图片，但是如果我们想单纯提取出mask并显示出来，本文将给您一些参考，本文没有写关于笔者如何提取mask的代码，如果需要请留言写下您的邮箱。

主要需要两个库，PIL和numpy。下面是代码，代码量很小，目前只是单个处理，后续我会贴上批量处理的代码。

# -*- coding: utf-8 -*-

import matplotlib.pyplot as plt
from PIL import Image
import numpy as np

img = Image.open("C:\\Users\\atom\\Desktop\image_mask.png")
img = Image.fromarray(np.uint8(img)*50)
img.save("C:\\Users\\atom\\Desktop\output_mask.png")

执行代码后即可在桌面上查看输出图片，发现可以看到你的mask了，结果如下图所示。
转化前：

转化后：

关于转化前mask为什么是“黑的”，是因为labelme标注类别是从0、1、2开始的，所以每个类别差距不大，所以显示出来看不出来，这个我参考了一位大神的说法，各位也可以看看。参考链接:https://blog.csdn.net/qq_15969343/article/details/79841446

本文重点是将mask显示出来，所以没有写提取工作，抱歉，需要请留言邮箱。
谢谢大家~