torch, PIL, matplotlib, opencv, seaborn.
pip install torch
pip install pillow
pip install matplotlib
pip install opencv-python
pip install seaborn
You can run "train.py", but it needs training dataset. In this project we use DUTS-TR.
python train.py
It needs the network weights. You can find it in the saved_models folder.
python utils.py
This project is based on U2Net created by Qin.X et al. original project address.