search

ZJULearning / resa

Implementation of our paper 'RESA: Recurrent Feature-Shift Aggregator for Lane Detection' in AAAI2021.
Apache License 2.0
175 stars 36 forks source link

A question about the "num_classes" in configs/tusimple.py line 87 #9

Closed minghongli233 closed 3 years ago

minghongli233 commented 3 years ago

This work is very nice! And I have one question to consult. As far as I know, the tusimple dataset have five lanes at most. But, the "num_classes" is seted to "6+1" in configs/tusimple.py line 87 . Could you tell me how to set the value? Thank you!

Turoad commented 3 years ago

Thank you. We pre-define the class in tusimple dataset, i.e., LEFT 3, 2, 1, RIGHT 3,2,1. Yes, the tusimple dataset have five lanes at most. Five lanes cases are: some images have LEFT 3, 2, 1, RIGHT 2, 1, but some have LEFT 2,1, RIGHT 3,2,1. RIGHT 3 and LEFT 3 are not the same class. Thus, the number of classes is 7, one is the background class.