search

koyeongmin / PINet_new

MIT License
166 stars 44 forks source link

How can I lightweight the PINet model? #35

Open suesuekkim opened 3 years ago

suesuekkim commented 3 years ago

I have some questions. Is there any solution to lightweight the PINet model? It seems good accuracy of detecting lanes, but the fps is quite low, so I wonder if there's any solution.

Thank you.

koyeongmin commented 3 years ago

If you want to get lightweight model, you can clip some hourglass network! (the method is described by README file) Also, you can modify network architecture. Modules other than the network can also cause slowdown, because I couldn't concentrate the code optimization. Some processing like clustering and spline regression consume quite a bit of time. Thank you!