Skip to content

Modified fork of Xuebin Qin's U-2-Net Repository. Used for demonstration purposes.

Notifications You must be signed in to change notification settings

shreyas-bk/U-2-Net

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

U^2-Net (U square net)

Modified version of U2Net used for demonstation purposes.

Xuebin Qin,
Zichen Zhang,
Chenyang Huang,
Masood Dehghan,
Osmar R. Zaiane and
Martin Jagersand.

Contact: xuebin[at]ualberta[dot]ca

Required libraries

Python 3.6
numpy 1.15.2
scikit-image 0.14.0
PIL 5.2.0
PyTorch 0.4.0
torchvision 0.2.1
glob

Usage

Check Demonstration

U-2-NET Paper: U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection

Original Repo: U-2-Net Github repo

References: X. Qin, Z. Zhang, C. Huang, M. Dehghan, O. R. Zaiane, and M. Jagersand, “U2-net: Going deeper with nested u-structure for salient object detection,” Pattern Recognition, vol. 106, p. 107404, 2020

About

Modified fork of Xuebin Qin's U-2-Net Repository. Used for demonstration purposes.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%