OpenDriveLab / OccNet

[ICCV 2023] OccNet: Scene as Occupancy
https://arxiv.org/abs/2306.02851
Apache License 2.0
566 stars 51 forks source link

The problem of occupancy prediction by bevformer #24

Open yanglong2000 opened 11 months ago

yanglong2000 commented 11 months ago

Hello,

occupancy_classes is set to 16 in the paper, but there are 17 labels in gt_occ (consider free space). So I wanted to ask whether the first (from the original) or the second is more appropriate.

occupancy_preds = occupancy_preds.view(-1, self.occupancy_classes) occupancy_preds = occupancy_preds.view(-1, self.occupancy_classes + 1)

I also found that both options can be trained and the initial loss of the first is much lower than that of the second.

600643356810c8d30d29e95ec0568ef f45e8dafad3e945b8b8924a628540db