Add U-2-Net Model - Githubissues

innat commented 11 months ago

Short Description

bg-removal

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 commented 11 months ago

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

innat commented 11 months ago

@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 commented 11 months ago

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

keras-team / keras-cv

Add U-2-Net Model #2008