U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection
Paper • 2005.09007 • Published
U²-Net
Input Shape:(512, 512, 3)
This is an implementation of U²-Net for image background removing purposes based on this paper. 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.
Dataset
Dichotomous Image Segmentation (DIS) is a dataset wich contains around 5k images (images + masks) and it's used in image segmenation model training.
You can find more about this dataset here.
Me:
Find me at GitHub: https://github.com/reidn3r
Find me at Linkedin: https://www.linkedin.com/in/reidner-adnan/
Project repository: GitHub repository
- Downloads last month
- 24
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support