Downloads · 30 days
26
2% of all-time downloads
u2net-image-rembg
reidn3r/u2net-image-rembg
u2net-image-rembg is a machine learning model from reidn3r. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for keras.
<pInput Shape:(512, 512, 3)</p <pThis is an implementation of U²-Net for image background removing purposes based on <a href="https://arxiv.org/pdf/2005.09007" target="blank"this paper</a. The U²-Net architecture is d…
All-time downloads
1.2K
Public
Repo size
765 MB
Likes
6
Public
Repo makeup
Click a slice to open those files.
.keras381 MB · 50%
At a glance
- Library
- keras
- Access
- Public
- Created
- Oct 8, 2024
- Updated
- Dec 6, 2025
- SHA
- af1a014b
Model card
From the Hugging Face model README
U²-Net
<p>Input Shape:(512, 512, 3)</p> <p>This is an implementation of U²-Net for image background removing purposes based on <a href="https://arxiv.org/pdf/2005.09007" target="_blank">this paper</a>. The U²-Net architecture is designed to capture both fine details and global context, making it highly effective for image segmentation tasks such as background removal.</p>Dataset
Dichotomous Image Segmentation (DIS) is a dataset wich contains around 5k images (images + masks) and it's used in image segmenation model training.<br> You can find more about this dataset <a href="https://github.com/xuebinqin/DIS" target="_blank">here.</a>
Me:
Find me at GitHub: <a href="https://github.com/reidn3r">https://github.com/reidn3r</a> <br> Find me at Linkedin: <a href="https://www.linkedin.com/in/reidner-adnan/">https://www.linkedin.com/in/reidner-adnan/</a> <br> Project repository: <a href="https://github.com/reidn3r/TensorFlow-U2-Net">GitHub repository</a>
- Library
- keras
- Languages
- pt, en
- Created
- Oct 8, 2024
- Updated
- Dec 6, 2025