search

huang-yh / SelfOcc

[CVPR 2024] SelfOcc: Self-Supervised Vision-Based 3D Occupancy Prediction
Apache License 2.0
273 stars 17 forks source link

How can I modify the number of classes #6

Open RikkiXu opened 8 months ago

RikkiXu commented 8 months ago

Thank you for your code, but I encountered problems when reproducing it. How can I modify the number of classes in NeuSHead? I used forward_occ to get the logits as 200x200x16x21, but I didn't find where to modify 21. I want to modify it to 16

huang-yh commented 3 months ago

https://github.com/huang-yh/SelfOcc/blob/7c2247479a09fe208a4782e5a399493d92d0b61b/config/nuscenes/nuscenes_occ.py#L350 You can modify the above line to change the number of predicted classes. Note that color_dims=3(RGB) + #classes.