U^2-Net (U square net)
Modified version of U2Net used for demonstation purposes.
U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection
Paper: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