用PaddleSeg的Matting预测 怎么得到的图片只有类似于蒙版的黑白图片
收藏
预测命令 python predict.py --config configs/modnet/modnet_mobilenetv2.yml --model_path modnet-mobilenetv2.pdparams --image_path 1.jpg --save_dir ./output/results
得到的是这样的图片
怎样才能得到正常的抠图后的前景图片
0
收藏
请登录后评论
写个简单的代码,把自己的原图和这个做个对比就解决了吧。
期待ing,嘿嘿嘿
https://aistudio.baidu.com/aistudio/projectdetail/2800577
这个项目里的bg_replace.py应该包括这部分代码,你可以自己看下。
怎样才能生成那种背景透明的png图片
所以你是想把人抠出来,然后生成背景透明的png图? 只是白底转透明背景的话百度有很多教程,你可以自己看看。
平台上没有这个功能的项目吗?要是没有的话,我周末写一个然后回你。
非常感谢,我就是想把人抠出来,然后生成背景透明的png图。别的项目抠图效果没有这个好,所以想用这个项目想把人抠出来,然后生成背景透明的png图,希望你用这个项目写一下,感谢。
png调色板了解一下,opencv就能做吧,blend图像
楼主应该要的是真值和原图的混合效果吧,可以参考这个博客:
https://blog.csdn.net/weixin_40522801/article/details/106597420
真值的颜色可以通过设置png的调色板改变。
有MASK后一行代码即可解决问题
img[mask==0] = 0
上述代码含义是将Mask中人以外的区域设置为黑色,如果是要保存PNG,遵循RGBA套用上述公式将A值设置为透明即可
输出效果没问题,mask白色区域是人像,黑色是背景
img是原图array,mask是mask的array
在维度一样的情况下(都是rgb或者都是灰度)
img[mask==0] = 0
img数组存起来就行了
cv2.imwrite(savepath,img)
Matting预测还有点不一样,matting每个像素点都有自己的阈值,所以需要单独进行处理。大家可以看新鲜出炉的项目:https://aistudio.baidu.com/aistudio/projectdetail/3416667 欢迎fork,点赞,留言交流。也欢迎看看我的其他项目。