Training question with classification labels customs data

Open 2713286758 opened this issue 3 years ago • 1 comments

I trained custom data according to the format of MOT17 label file, using classification labels from 1 to 4, but I got an error.It illustrates "Assertion idx_dim>=0 && idx_dim<index_size && "index out of bounds" failed.", in models/deformable_detrtrack_train.py line 382 "target_classes_onehot.scatter_(2, target_classes.unsqueeze(-1), 1)".Does anyone have the same bug as me?

May 06 '22 09:05 2713286758

I have sloved problem above.

May 13 '22 02:05 2713286758