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