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

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

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