mirror of
https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net
synced 2025-08-20 01:39:05 +08:00
64 lines
2.3 KiB
Markdown
64 lines
2.3 KiB
Markdown
# U^2-Net
|
|
|
|
The code for our newly accepted paper in Pattern Recognition 2020:
|
|
## U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection, [Xuebin Qin](https://webdocs.cs.ualberta.ca/~xuebin/), [Zichen Zhang](https://webdocs.cs.ualberta.ca/~zichen2/), [Chenyang Huang](https://chenyangh.com/), [Masood Dehghan](https://sites.google.com/view/masooddehghan), [Osmar R. Zaiane](http://webdocs.cs.ualberta.ca/~zaiane/) and [Martin Jagersand](https://webdocs.cs.ualberta.ca/~jag/).
|
|
|
|
__Contact__: xuebin[at]ualberta[dot]ca
|
|
|
|
## U^2-Net Results (173.6 MB)
|
|
|
|
|
|
|
|
|
|
## Our previous work: [BASNet (CVPR 2019)](https://github.com/NathanUA/BASNet)
|
|
|
|
## 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
|
|
1. Clone this repo
|
|
```
|
|
git clone https://github.com/NathanUA/U-2-Net.git
|
|
```
|
|
2. Download the pre-trained model [u2net.pth (173.6 MB)](https://drive.google.com/file/d/1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ/view?usp=sharing) or [u2netp.pth (4.7 MB)](https://drive.google.com/file/d/1rbSTGKAE-MTxBYHd-51l2hMOQPT_7EPy/view?usp=sharing) and put it into the dirctory './saved_models/u2net/' and './saved_models/u2netp/'
|
|
|
|
3. Cd to the directory 'U-2-Net', run the train or inference process by command: ```python u2net_train.py```
|
|
or ```python u2net_test.py``` respectively. The 'model_name' in both files can be changed to 'u2net' or 'u2netp' for using different models.
|
|
|
|
We also provide the predicted saliency maps ([u2net results](https://drive.google.com/file/d/1mZFWlS4WygWh1eVI8vK2Ad9LrPq4Hp5v/view?usp=sharing),[u2netp results](https://drive.google.com/file/d/1j2pU7vyhOO30C2S_FJuRdmAmMt3-xmjD/view?usp=sharing)) for datasets SOD, ECSSD, DUT-OMRON, PASCAL-S, HKU-IS and DUTS-TE.
|
|
|
|
|
|
## U^2-Net Architecture
|
|
|
|
|
|
|
|
|
|
## Quantitative Comparison
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Qualitative Comparison
|
|
|
|
|
|
|
|
|
|
## Citation
|
|
```
|
|
@InProceedings{Qin_2020_PR,
|
|
author = {Qin, Xuebin and Zhang, Zichen and Huang, Chenyang and Dehghan, Masood and Zaiane, Osmar and Jagersand, Martin},
|
|
title = {U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection},
|
|
booktitle = {Pattern Recognition},
|
|
year = {2020}
|
|
}
|
|
```
|