Add U-2-Net Model

[innat]

Short Description

• U2-Net (U square net) published in Pattern Recognition 2020
• Awarded the 2020 Pattern Recognition BEST PAPER AWARD
• Integrated in iOS App 3D Photo Creator
• Integrated in iOS App Portrait Drawing based U2-Net portrait generation model.
• Integrated in iOS App Lensto, (Demo Video), based on U2-Net.
• Integrated in Hotpot.ai for art design.
• Image editing app Pixelmator pro uses U2-Net as one of its background removal models.
• Integrated in background removal webapp developed by Изатоп Василий.
• Cited by 825 [until now]

Papers

Article: https://arxiv.org/abs/2005.09007

Existing Implementations

Official PyTorch implementation: https://github.com/xuebinqin/U-2-Net Unofficial: keras: https://github.com/chjort/u2net

Other information

Succeeding work of BASNet (same author).

[cosmo3769]

Hey @innat, I would love to take this up. Thank you.

[innat]

@cosmo3769 The author of BASNet or U2Net has published their lastest work, DIS, all of these models serve the same purpose (segmentation) and similar architecture in most part. If you're up for working, please choose the latest model (i.e., DIS).

[cosmo3769]

@innat @ianstenbit Okay, I would love to add DIS model if it aligns with keras-cv now. Suggestions please?

[frostbyte012]

@innat I would like to contribute as well! Is it still available to work on?