xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.52k stars 1.47k forks source link

u2net_train.py returning ValueError #376

Closed rajmanna-dev closed 8 months ago

rajmanna-dev commented 8 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-

298676491-3ac92c00-ad77-42a5-b823-2512937fa52d

I am using python 3.9 for this project. I'm a beginner in python.. It was 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 😢.

Muzi010 commented 6 months ago

你解决了么?我和你遇见了相同的问题,能否讨论一下