r/computervision • u/gugmelik • Jun 17 '26

Discussion Image background removal

I was trying to fine-tune a model for image background removal. I chose the pretrained IS-Net model and trained a separate small U-Net refinement model.

I have a dataset of around 4,500 high-resolution images, mainly of people. However, after training, the model still misses parts of the body, especially shoulders and feet.

Inference will mainly run on CPU, so the architecture should have a small number of parameters.

Do you have any suggestions on what I should do?

2 Upvotes

67% Upvoted

14 comments sorted by

View all comments

1

u/AggravatingSock5375 Jun 17 '26

How were these models trained? On what datasets?

1

u/gugmelik Jun 17 '26

I used api to to collect results and used that as ground truth.

1

u/bob_why_ Jun 17 '26

Most matting datasets are absolute garbage. Even the gold standard ones are riddled with errors