VMA icon indicating copy to clipboard operation
VMA copied to clipboard

Published 20 hours ago • verified publisher icon hustvl

About attr_mlp GT

Open bitwangdan opened this issue 2 years ago • 2 comments

image Hi, in sd_driviing_line_dataset.py, if instance_class == "lane", the category of curb is 6, but the curb is six-category branch, index=[0, 1, 2, 3, 4, 5], Wouldn't this index out of bounds?

bitwangdan avatar Sep 13 '23 11:09 bitwangdan

@zyc10ud

bitwangdan avatar Sep 14 '23 03:09 bitwangdan

You can refer to the source code of the focal loss in mmdet. https://github.com/open-mmlab/mmdetection/blob/f78af7785ada87f1ced75a2313746e4ba3149760/mmdet/models/losses/focal_loss.py#L238C21-L239C53

zyc10ud avatar Sep 14 '23 12:09 zyc10ud