search

Ezaldeen99 / BackgroundRemoval

Background Removal written with swift using u2net model
Apache License 2.0
204 stars 34 forks source link

Background removal on portrait images #3

Closed jonasfunk closed 2 years ago

jonasfunk commented 2 years ago

Hi, thanks for the great work you've done with this library! Images passed to the model are cropped to be square. I've tried handing it portrait images, but then it crashes. Is it only able to handle square images?

jonasfunk commented 2 years ago

I've found out that the model is trained on square images.