search

hustvl / VMA

A general map auto annotation framework based on MapTR, with high flexibility in terms of spatial scale and element type
MIT License
210 stars 22 forks source link

About attr_mlp GT #5

Open bitwangdan opened 1 year ago

bitwangdan commented 1 year ago

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 commented 1 year ago

@zyc10ud

zyc10ud commented 1 year ago

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