Skip to content

Norod/U-2-Net-StyleTransfer

Repository files navigation

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:

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)

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

Releases

No releases published

Packages

No packages published

Languages