图像分割时的label设置

  1. 若待分割的目标物的种类为n,那么label的值最大为n-1。
    比如我要分割4类物体,那么我的label就应当设置为0,1,2,3。
    设置的方法为:
    (1)label为单通道有值时,比如只是R通道有值,那么可以使用PIL中的.convert(‘P’)将label图片转为通过调色板与三通道RGB图像对应的单通道图。
    (2)label为三个通道都有值的情况,比如目标物一为[128,0,0],目标物二为[0,128,0],目标物三为[0,0,128],目标物四为[0,0,0]。那我们可以先除以128后,给每个通道赋值,然后将通道相加。
# r = 1 , g=2, b=3
np_img = np.array(pil_img)
np_img /= 128
label = 1*np_img[:,:,0] + 2*np_img[:,:,1] + 3*np_img[:,:,2]
return label  

2.numpy.unique()可以滤除numpy矩阵中重复的值,并顺序排好矩阵中的值。
可以用它查看我们设置的label图片是否正确。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值