图像分割标注工具labelme各个版本改变标注颜色

labelme没有draw文件了,实现标注是调用imgviz库的label文件,所以改一下label文件的相应代码就可以实现了。
1.labelme的版本为4.5.6
具体位置在conda虚拟环境位置下的envs/labelme/lib/python3.6/site-packages/imgviz/label.py中,将

 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))

改为

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的版本为4.5.6。
2.labelme的版本为5.0.1
具体位置在conda虚拟环境位置下的envs/labelme/Lib/site-packages/imgviz/label.py中,将

	i = np.repeat(i[:, None], 8, axis=1)
    i = np.right_shift(i, np.arange(0, 24, 3)).astype(np.uint8)
    j = np.arange(8)[::-1]
    r = np.bitwise_or.reduce(np.left_shift(bitget(i, 0), j), axis=1)
    g = np.bitwise_or.reduce(np.left_shift(bitget(i, 1), j), axis=1)
    b = np.bitwise_or.reduce(np.left_shift(bitget(i, 2), j), axis=1)

修改为

i = np.repeat(i[:, None], 8, axis=1)
    i = np.right_shift(i, np.arange(0, 24, 3)).astype(np.uint8)
    j = np.arange(8)[::-1]
    r = np.bitwise_or.reduce(np.left_shift(bitget(i, 0), j), axis=1)
    g = np.bitwise_or.reduce(np.left_shift(bitget(i, 1), j), axis=1)
    b = np.bitwise_or.reduce(np.left_shift(bitget(i, 2), j), axis=1)
    r=np.full_like(r,255)#这里255,0,0三个数字可以随便改,改成你所指定的颜色
    g=np.full_like(g,0)
    b=np.full_like(b,0)

修改后重启动

3.指定不同标签的不同颜色(版本5.0.1)

具体位置在conda虚拟环境位置下的envs/labelme/Lib/site-packages/imgviz/label.py中,将

	r = np.bitwise_or.reduce(np.left_shift(bitget(i, 0), j), axis=1)
    g = np.bitwise_or.reduce(np.left_shift(bitget(i, 1), j), axis=1)
    b = np.bitwise_or.reduce(np.left_shift(bitget(i, 2), j), axis=1)

    cmap = np.stack((r, g, b), axis=1).astype(np.uint8)
	r = np.bitwise_or.reduce(np.left_shift(bitget(i, 0), j), axis=1)
    g = np.bitwise_or.reduce(np.left_shift(bitget(i, 1), j), axis=1)
    b = np.bitwise_or.reduce(np.left_shift(bitget(i, 2), j), axis=1)
    
    cmap = np.stack((r, g, b), axis=1).astype(np.uint8)
    cmap[1, :] = [0,255,0]  # 种类1的颜色
    cmap[2, :] = [255,0,0]  # 种类2的颜色,可以改变指定颜色或添加更多指定种类
  • 17
    点赞
  • 60
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 19
    评论
评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不是吧这都有重名

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值