/U2NET-training

Scripts for training U2NET model

Primary LanguagePython

How to use

python u2net_train.py --data-path '/content/data' \
--save-path '/content/drive/MyDrive/U2NET-training/logs' \
--epoch 100 --train-batch 12 --worker 2 \
--model 'u2net' --print-frq 50