wzzheng / OccWorld

[ECCV 2024] 3D World Model for Autonomous Driving
https://wzzheng.net/OccWorld/
Apache License 2.0
382 stars 25 forks source link

About Dataset #6

Closed VitaLemonTea1 closed 11 months ago

VitaLemonTea1 commented 12 months ago

Hi,

Thanks for releasing code. Do I need to download the whole nuscene dataset, or just need Lidarseg.

Thanks!

chen-wl20 commented 12 months ago

Thank you for your interest! I possess the complete dataset, yet it seems that all I require is the gts from Occ3D. And other annotations are available in the pkl file.

VitaLemonTea1 commented 12 months ago

Thanks for your reply. I see the Occ3D page, and see that gts is not in the code space. Is it in the nuScenes-lidarseg?

chen-wl20 commented 12 months ago

Thank you for your attention! Occ3D is a dataset focused on 3D occupancy. The file structure is as follows, and you can easily load the .npz files using numpy for your applications. For more details please follow the Occ3d page. ├── gts
| ├── [scene_name] | | ├── [frame_token] | | | └── labels.npz

VitaLemonTea1 commented 11 months ago

Thanks,I check the webpage and download it! After running train.py , it got another problem, AttributeError: 'ConfigDict' object has no attribute 'load_from'. I have checked the issues of mmengine, but there is no help. I wander if you can help me fix it.

chen-wl20 commented 11 months ago

Thank you for your continued attention! This bug can be solved by adding the vqvae model path using load_from in the config file. We have update the code and the config files just now, which may contribute to your problem!

VitaLemonTea1 commented 11 months ago

Thanks for your help. It has been runned successfully

VitaLemonTea1 commented 11 months ago

I read the paper and see that you and your colleges trained the model with 8x4090. Could you please tell me how long you spend on training.