tensorflow - 如何实现强大的背景去除？

Question

我发现基于深度学习的方法(例如 1)比非基于深度学习的方法(例如 2，使用 OpenCV)健壮得多。

https://www.remove.bg

How do I remove the background from this kind of image?

在 OpenCV 示例中，Canny 用于检测边缘。但是这一步可能对图像非常敏感。轮廓检测可能会以错误的轮廓结束。也很难确定应该保留哪些轮廓。

如何实现稳健的深度学习方法？有什么好的示例代码吗？谢谢。

Answer 1

为此，您需要使用 Unet。你可以在github上搜索。
Unet 转换为:I->I。
图像的空间将成为图像(相同或相似大小)。

您需要说 10.000 张删除了 bg 的图像。人、(长发人)、猫、汽车、鞋子、T 恤等。

所以你在所有这些图像上设置不同的背景作为源和预测应该是去除背景的图像。

你也可以做一个 segmentation model当您找到前景时，您可以删除背景。