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
295 stars 74 forks source link

A light-weight backbone? #9

Closed xindongzhang closed 4 years ago

xindongzhang commented 4 years ago

Thanks for sharing your great job with us. Is it possible to replace the feature extraction backbone with a smaller backbone? Such as MobileNet, ShuffleNet, and SqueezeNet, etc.

lucastabelini commented 4 years ago

Yes. In this part of the code, the backbone is chosen. As it can be seen, it should be fairly easy to add a new one.

xindongzhang commented 4 years ago

Yes. In this part of the code, the backbone is chosen. As it can be seen, it should be fairly easy to add a new one.

Thanks for your reply