IndexError: index 0 is out of bounds for dimension 0 with size 0 during train

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Other

1.7k stars 307 forks source link

IndexError: index 0 is out of bounds for dimension 0 with size 0 during train #188

Open azadehashouri opened 3 years ago

azadehashouri commented 3 years ago

Hi, I'd like to train SOLO on my custom dataset which contains one class. Then I got this error during training the model. I'll appreciate your help. Many thanks Capture

wangzehui20 commented 3 years ago

you can add 'background' in your CLASSES

t-nakatani commented 2 years ago

I have met the same problem. How did you solve this ?