labelme制作数据集改变掩码的颜色

草莓味的风

于 2020-07-08 13:06:15 发布

阅读量6.4k

点赞数 5

文章标签：数据集制作

本文链接：https://blog.youkuaiyun.com/hr786250678/article/details/107203354

版权

labelme生成的mask默认是黑色和红色的，如何将其修改成黑白二值的呢？

首先找到labelme的安装路径，labelme是通过anaconda虚拟环境运行的，所以其安装路径在在anaconda环境下的Lib/site-packages/目录下。

找到draw.py文件，在如下处修改：

if i == 1:
    r = 255
    g = 255
    b = 255
else:
    r = np.bitwise_or(r, (bitget(id, 0) << 7-j))
    g = np.bitwise_or(g, (bitget(id, 1) << 7-j))
    b = np.bitwise_or(b, (bitget(id, 2) << 7-j))

再次运行 labelme_json_to_dataset <文件名>.json，就会发现mask变成黑白的了。