danielgatis / rembg

Rembg is a tool to remove images background
MIT License
15.46k stars 1.76k forks source link

u2net_train.py returning ValueError #574

Closed rajmanna-dev closed 5 months ago

rajmanna-dev commented 5 months ago

Can anyone help me to fix this: I'm trying to train u2net with DUTS dataset for my personal python project (removing complex background from images) u2net_test.py run successfully with DUTS dataset but u2net_train.py is returning error like this- Screenshot 2024-01-22 232247

I am using python 3.9 for this project. I'm a beginner in python.. It is very helpful for me if anyone teach me step by step how to trained u2net with custom dataset like DUTS.. I try almost very thing I can 😢.