search

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 the lane_width #20

Closed pocky111 closed 11 months ago

pocky111 commented 1 year ago

Hello, I have been reading your code recently, and I don't quite understand why lane_width= 15/800 in the calculation of loss function. Could you please explain it to me?

hirotomusiker commented 11 months ago

lane_width is the half-width to form virtual lane markers and calculate IoU. In LineIoU (CLRNet), it's called length and set to 15 in the image resolution of 800. So we set it to 15/800 in the relative coordinate system (0 to 1.0) as the default value. We actually set lane_width to 7.5 / 800 to better represent the lane width for evaluation for CLRerNet. We will include the settings in our training codebase. Thank you!