U2-Net: U Square Net
Description:
Improving u2Net performance using classical alpha metting techniques.
BaseCode:
pymatting: https://github.com/pymatting/pymatting u2Net: https://github.com/xuebinqin/U-2-Net/blob/master/u2net_train.py
Setup:
- Install All the Dependencies ( its recommended to create a new env.)
pip install -r requirements.txt
-
Download DUTS Dataset from
.
-
Put DUTS-TR inside
train_data/DUTS/ -
you can also download the pretrained models from
and put it inside
saved_modelsfolder -
run the training script:
python u2net_train.py
Architecture:
Results:
