将Matting结果转化为二分割结果

matting问题本质上作为一个回归问题,其生成的alpha matte包含的是每个像素属于前景的“程度”;而二分割则是分类问题,每个像素要么属于前景,要么属于背景。

可以设置一个阈值,认为 α \alpha α值超过某个给定阈值则认为其为前景,小于该阈值则认为其为背景,从而达到问题的转化。

比如我们有个matting结果如下(图源alphamatting.com数据集):
在这里插入图片描述
α \alpha α值非零的像素点都认为是前景(在这里为人像),结果如下:
在这里插入图片描述
代码:

from skimage import io

img = io.imread("GT08.png")
idx = (img > 0)
img[idx] = 255
io.imsave('res.png', img)

利用的是numpy的布尔数组索引。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值