Anyway to fix what U2NET removes?

[FASTANDEXTREME]

Sometimes it removes objects that were not supposed to be removed. For example,

In this image, gun also got removed but it was part of the image and not background.

[zhefangbyte]

I was also struggling with such kind of requirement, I found these three repositories recently:

1. https://github.com/geekyutao/Inpaint-Anything
2. https://github.com/advimman/lama
3. https://github.com/facebookresearch/segment-anything

They provide implementation to achieve image segmentation and inpainting with mask. Sometimes it has a better result than rembg, sometimes it not.

[Renvect]

Been dealing with this for some time, and think we got solution for this but requires combo with OpenCV to perform dual analysis and creation of composite mask based on this analysis with couple more tricks. So far it successfully restores lost contours and details within. We are looking to enhance it and expand to variety of use cases. I will update soon with results after we run some additional tests as many struggle with this part.

[FASTANDEXTREME]

will be looking forward for it, please update if you have any progress. Thank you.

…

On Mon, Jan 29, 2024 at 9:16 PM RAZ ***@***.***> wrote: Been dealing with this for some time, and think we got solution for this but requires combo with OpenCV to perform dual analysis and creation of composite mask based on this analysis with couple more tricks. So far it successfully restores lost contours and details within. We are looking to enhance it and expand to variety of use cases. I will update soon with results after we run some additional tests as many struggle with this part. — Reply to this email directly, view it on GitHub <#566 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/APYLRSXIEIXGDGAY6KYNQCDYQ674BAVCNFSM6AAAAABBZ3FG2OVHI2DSMVQWIX3LMV43SRDJONRXK43TNFXW4Q3PNVWWK3TUHM4DEOBVGQZTI> . You are receiving this because you authored the thread.Message ID: ***@***.***>