search

liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.
BSD 3-Clause "New" or "Revised" License
645 stars 129 forks source link

Hello，How to get the infering time for an img in test.py #68

Open Timooooooooooo opened 2 years ago

Timooooooooooo commented 2 years ago

I can not find some sentences like"model(img)"...Look forward for your answer！

liuruijin17 commented 2 years ago

Hi， I suppose https://github.com/liuruijin17/LSTR/blob/c8fe59a1d8e1c456c982c9e954ee660ae9116b49/test/tusimple.py#L118 is the sentence you need.