pytorch U2Net的背景去除太强

plicqrtu  于 2023-01-05  发布在  其他
关注(0)|答案(1)|浏览(181)

我已经成功地使用U2Net在终端上去除了图像的背景,我也使用了this repo的漂亮界面,以一种更简单的方式做了同样的事情,并验证了结果的相似性。然而,我的问题是,背景去除对于这样的图像来说太强了:

我得到了以下结果(即 Package 也被移除):

如果我上传Foco clipping website的图片并选择Type=='Graphic',我会得到完全相同的结果。这意味着网站使用相同的算法来去除图形类型图片的背景。然而,如果我选择Type=='Product',那么结果如下所示,这正是我想要的:

有人知道该怎么做才能得到同样的结果吗?

58wvjzkj

58wvjzkj1#

1.你应该锐化这图象第一(使用强锐化)

/* I've removed the code because I am not sure if it has some portion copied from other open-sourced codes, I am using it in my apps but I can't remember if it is quoted from other source, so I've removed it, also it could be auto-completed using github copilot */

1.使用U2Net去除锐化图像的背景

1.使用步骤(2)的结果作为掩模

1.使用步骤(3)的掩模从原始图像中提取想要的结果

注意:这是一个非常快速的例子,你可以进一步完善它

相关问题