首页 Paddle框架 帖子详情
用PaddleSeg的Matting预测 怎么得到的图片只有类似于蒙版的黑白图片
收藏
快速回复
Paddle框架 问答深度学习 2044 11
用PaddleSeg的Matting预测 怎么得到的图片只有类似于蒙版的黑白图片
收藏
快速回复
Paddle框架 问答深度学习 2044 11

预测命令  python predict.py --config configs/modnet/modnet_mobilenetv2.yml --model_path modnet-mobilenetv2.pdparams --image_path 1.jpg --save_dir ./output/results

得到的是这样的图片

 

怎样才能得到正常的抠图后的前景图片

 

0
收藏
回复
全部评论(11)
时间顺序
十进制到二进制
#2 回复于2022-01

写个简单的代码,把自己的原图和这个做个对比就解决了吧。

0
回复
JavaRoom
#3 回复于2022-01
写个简单的代码,把自己的原图和这个做个对比就解决了吧。

期待ing,嘿嘿嘿

0
回复
十进制到二进制
#4 回复于2022-01

https://aistudio.baidu.com/aistudio/projectdetail/2800577

这个项目里的bg_replace.py应该包括这部分代码,你可以自己看下。

0
回复
QQ360543679
#5 回复于2022-01
https://aistudio.baidu.com/aistudio/projectdetail/2800577 这个项目里的bg_replace.py应该包括这部分代码,你可以自己看下。

怎样才能生成那种背景透明的png图片

0
回复
十进制到二进制
#6 回复于2022-01
怎样才能生成那种背景透明的png图片

所以你是想把人抠出来,然后生成背景透明的png图? 只是白底转透明背景的话百度有很多教程,你可以自己看看。

平台上没有这个功能的项目吗?要是没有的话,我周末写一个然后回你。

0
回复
QQ360543679
#7 回复于2022-01
所以你是想把人抠出来,然后生成背景透明的png图? 只是白底转透明背景的话百度有很多教程,你可以自己看看。 平台上没有这个功能的项目吗?要是没有的话,我周末写一个然后回你。

非常感谢,我就是想把人抠出来,然后生成背景透明的png图。别的项目抠图效果没有这个好,所以想用这个项目想把人抠出来,然后生成背景透明的png图,希望你用这个项目写一下,感谢。

0
回复
悠哈大飞
#8 回复于2022-01

png调色板了解一下,opencv就能做吧,blend图像

0
回复
悠哈大飞
#9 回复于2022-01

楼主应该要的是真值和原图的混合效果吧,可以参考这个博客:

https://blog.csdn.net/weixin_40522801/article/details/106597420

真值的颜色可以通过设置png的调色板改变。

0
回复
GT
#10 回复于2022-01

有MASK后一行代码即可解决问题

img[mask==0] = 0

上述代码含义是将Mask中人以外的区域设置为黑色,如果是要保存PNG,遵循RGBA套用上述公式将A值设置为透明即可

 
1
回复
奔向未来的样子
#11 回复于2022-01

输出效果没问题,mask白色区域是人像,黑色是背景

img是原图array,mask是mask的array

在维度一样的情况下(都是rgb或者都是灰度)

img[mask==0] = 0

img数组存起来就行了

cv2.imwrite(savepath,img)

0
回复
十进制到二进制
#12 回复于2022-01

Matting预测还有点不一样,matting每个像素点都有自己的阈值,所以需要单独进行处理。大家可以看新鲜出炉的项目:https://aistudio.baidu.com/aistudio/projectdetail/3416667 欢迎fork,点赞,留言交流。也欢迎看看我的其他项目。

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户