Bootstrap

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

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变成黑白的了。

 

 

;