Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Tweaking to improve results #236

Closed
bullmoose20 opened this issue May 28, 2022 · 6 comments
Closed

Tweaking to improve results #236

bullmoose20 opened this issue May 28, 2022 · 6 comments

Comments

@bullmoose20
Copy link

If we see anomalies when removing backgrounds, are there tweaks we can make? I am mostly trying to remove backgrounds on people headshots and noticed that sometimes a shoulder or everything below the neck disappears...

@bullmoose20
Copy link
Author

Before:
Josh Gad
Karl Urban

After (using rembg i):
Josh Gad
Karl Urban

After (using rembg i a -ae):
Josh Gad-ae
Karl Urban-ae

@bullmoose20
Copy link
Author

Additionally, maybe there is a way for me to specify this somehow?
u2net_cloth_seg (download, source): A pre-trained model for Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body.

And I think almost all are "Upper body"

@bullmoose20
Copy link
Author

rembg i -m u2net_human_seg

--help does not show the -m option
README.md does not specify that there is a -m option... I stumbled on this...

@bullmoose20
Copy link
Author

While this is a better option and I get better results, I would still like to know how I can contribute to the model to improve it. I have about 200 people headshots before and after background removal and would like to know how I can contribute to the community?

@ArielMAJ
Copy link

I ran into this same thing and u2net_human_seg helped too.

As for improving upon the model, I'm pretty sure you should be able to train your own model (or train u2net_human_seg with your own images) and use it. Pretty sure it isn't an easy task either. GL!

@alcinopitchprint
Copy link

The training process is explained here by @endh1337 , #193 (comment)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants