#97 · Primary category: Computer Vision

U-2-Net

computer-vision deep-learning image-background-removal image-processing image-segmentation u-2-net u2net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Project last updated:06/26/24

GitHub Stars

9.9K

Forks

1.6K

Contributors

13

License

Apache-2.0

Why we included this project

U-2-Net isolates the most salient object in an image and returns a clean cutout, which makes it a dependable first stop for background removal and portrait matting. The repository accompanies the Pattern Recognition 2020 paper and ships pretrained weights plus the scripts used to train, test, and run the network, so you can move from a raw photo to a masked result without reimplementing anything. Teams building photo editors or automated product-image tooling have plugged these weights in directly, and because the model keeps full-resolution output without the heavy preprocessing other segmentation approaches require, integration stays fairly simple. It is research code aimed at a specific task rather than a broad pipeline, so expect to wire it into your own application instead of getting a one-click product.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category