zzzzzzpc / portrait-segmentation

使用U-net网络结构进行的人像分割代码,部署于TensorFlow Serving.
2 stars 0 forks source link

请问效果如何? #1

Open yangjian1218 opened 3 years ago

yangjian1218 commented 3 years ago

效果如何,人像是否容易身体切分空缺, 黑色连接背景是否可以有效的分割出去?

zzzzzzpc commented 3 years ago

在测试集上有96%左右的Miou,如果只有半身效果还可以,可以看看static/image/里面的portrait和result两个图片

yangjian1218 commented 3 years ago

推理代码看不到,所以不懂的推理结果是什么形式,我看你对掩码直接二值化了, mask[mask 255 > 128] = 255 mask[mask 255 <= 128] = 0, 这样会显得抠图不自然, 建议你归一化后,使用公式 :原图mask+ 底图(1-mask)

zzzzzzpc commented 3 years ago

我想你说的“推理代码”是有关model和train的代码,我并没有上传相关代码。如果你喜欢的话,我可以给你有关的数据集和标注集以及训练代码,也许你能得到更好的结果,不过作为实时分割的探索,我已经找到了更好的方案,我使用了mediapipe构建实时分割app,效果也很不错,喜欢的话可以试试:https://github.com/zzzzzzpc/piepic

yangjian1218 commented 3 years ago

谢谢,这个模型输出是(512,512,2) 这是啥图?

---原始邮件--- 发件人: @.> 发送时间: 2021年5月15日(周六) 下午2:24 收件人: @.>; 抄送: @.**@.>; 主题: Re: [zzzzzzpc/portrait-segmentation] 请问效果如何? (#1)

我想你说的“推理代码”是有关model和train的代码,我并没有上传相关代码。如果你喜欢的话,我可以给你有关的数据集和标注集以及训练代码,也许你能得到更好的结果,不过作为实时分割的探索,我已经找到了更好的方案,我使用了mediapipe构建实时分割app,效果也很不错,喜欢的话可以试试:https://github.com/zzzzzzpc/piepic

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.