hirotomusiker / CLRerNet

The official implementation of "CLRerNet: Improving Confidence of Lane Detection with LaneIoU"
Apache License 2.0
174 stars 18 forks source link

compute laneiou #29

Closed zyk516 closed 3 weeks ago

zyk516 commented 9 months ago

Hello, according to https://github.com/hirotomusiker/CLRerNet/blob/main/configs/clrernet/base_clrernet.py#L53, iou_dynamick uses lane_width=7.5 / 800 and use_pred_start_end=False, but iou_cost uses lane_width=30 / 800 and use_pred_start_end=True.I want to know why they use different settings. Could you please explain it to me?

hirotomusiker commented 9 months ago

Thank you, we employ wider virtual width for the IoU cost just to balance with the classification cost. https://github.com/hirotomusiker/CLRerNet/blob/main/libs/core/bbox/assigners/dynamic_topk_assigner.py#L180