This is for researchers who are supported with multi-gpus!

** For this particular process you will need pytorch lightning

• Pretrained weights (DUTS-TR):
  checkpoints : u2net.ckpt
  state_dict: u2net.pth
• I have updated inference.py code to test u2net
  

Python 3.8
Pytorch 1.12.0+cu102
PyTorch Lightning 1.8.6
Numpy 1.23.5
Opencv-Python 4.6.0
Albumentations 1.3.0
wandb 0.13.8 (excluded in requirements)

1. Conda

conda create -n <env-name> python=3.8

2. pip

pip install -r requirements.txt

1. arguments

2. Script
   ** I made it easier to change arguments by changing in python script

python train_u2net.py