U²-Net: U Square Net

Modified for paired image training of style transfer

This is an unofficial repo making use of the code which was made available by the original author team of U²-Net (U square net) published in Pattern Recognition 2020:

U²-Net: Going Deeper with Nested U-Structure for Salient Object Detection

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

Before you clone this repo

Make sure you have git lfs installed and configured, so the actual model files will be downloaded rather than just a reference pointer to their location

Inference

See u2net_test.py

input	u2netp (4.9 MB)	u2net (176.5 MB)

Training

See u2net_train.py

Pre-Trained models

Available here

Citation data of the original Repo

@InProceedings{Qin_2020_PR,
title = {U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection},
author = {Qin, Xuebin and Zhang, Zichen and Huang, Chenyang and Dehghan, Masood and Zaiane, Osmar and Jagersand, Martin},
journal = {Pattern Recognition},
volume = {106},
pages = {107404},
year = {2020}
}