how to set N when testing

liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.

BSD 3-Clause "New" or "Revised" License

647 stars 130 forks source link

how to set N when testing #55

Closed tomorrow00 closed 3 years ago

tomorrow00 commented 3 years ago

Hi, great work! I'd like to ask that how to set the lanes number N when testing. N is fixed during training, will the testing process output the same N with training?

liuruijin17 commented 3 years ago

Hi, thanks a lot!

Yes, LSTR outputs N ‘lanes’ during testing. Those 'lanes' whose categories are 1 (0: background, 1: lane) would be the final lane predictions.

tomorrow00 commented 3 years ago

Hi, thanks a lot!

Yes, LSTR outputs N ‘lanes’ during testing. Those 'lanes' whose categories are 1 (0: background, 1: lane) would be the final lane predictions.

Appreciate that, it helps me so much!