search

lucastabelini / PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)
https://arxiv.org/abs/2004.10924
MIT License
301 stars 75 forks source link

poly_weight #58

Closed Hank880223 closed 2 years ago

Hank880223 commented 2 years ago

Hello, what does the poly_weight parameter mainly do? If I want to train a polynomial regression of order 4 or 5, will the parameter poly_weight affect the training?

Thank you

lucastabelini commented 2 years ago

The poly_weight defines the weights for each polynomial component in the loss function, i.e., higher weights have a higher impact on the loss.

Hank880223 commented 2 years ago

I understand it as higher poly_weight, can the predicted lane points be closer to the real lane points?