bradyz / cross_view_transformers

Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)
MIT License
531 stars 81 forks source link

How to train this for segmenting more than 2 classes? #43

Open GoroYeh56 opened 1 year ago

GoroYeh56 commented 1 year ago

Hi,

I have two specific questions:

  1. For label generation, I only got labels like this: Only white and black colors. I want to generate labels for more than 2 classes like the results shown in Figure 5 on the paper. What parameters should I change to achieve this? visibility_448fb51aeb4941c0abb029686572c14a

  2. For training a model that produces Figure 5. on the paper, what parameters should I change?

Thank you so much!

linlion0311 commented 1 year ago

me too