WXinlong / SOLO

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

azadehashouri opened this issue · comments

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

you can add 'background' in your CLASSES

commented

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