This is a U-2-Net Code Snippet for U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection.
This implementation is based on official's repo, which can also make use of pre-trained model u2net.pth (176.3 MB) or u2netp.pth (4.7 MB).