koyeongmin / PINet

MIT License
207 stars 56 forks source link

is it possible to train ten lanes? #3

Closed lfxx closed 4 years ago

lfxx commented 4 years ago

Thanks for your execllent work! When i read your source code,i realize that your code can train at most 6 lanes,can i train ten lanes? Do i need to modify net structure?hope for your kindly reply:)@koyeongmin

koyeongmin commented 4 years ago

Our network can train on the arbitrary number of lanes. In dataloader, the code use at most 6 lanes, but this is just for improving accuracy on tuSimple dataset. If you do not divide dataset according to the number of lanes, you can train 10 lanes.

lfxx commented 4 years ago

thanks!