U2-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 U2-Net (U square net) published in Pattern Recognition 2020:

U2-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

Inference

See u2net_test.py

input

u2netp (4.9 MB)

u2net (176.5 MB)

input u2netp u2net
input u2netp u2net

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}
}

GitHub

View Github