Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training on nuscenes dataset #107

Open
thom966 opened this issue Apr 24, 2022 · 1 comment
Open

Training on nuscenes dataset #107

thom966 opened this issue Apr 24, 2022 · 1 comment

Comments

@thom966
Copy link

thom966 commented Apr 24, 2022

Hello CaDDN author, thank you for your great work.

I am trying to train the model using nuscenes dataset. So, I have modified the config yaml file but I received an error
/backbones_3d/ffe/depth_ffe.py", line 51, in forward images = batch_dict["images"] KeyError: 'images'

It seems like the dataloader extracts batch of data for lidar only, not camera front view.

array(['n008-2018-08-28-16-43-51-0400__LIDAR_TOP__1535489298547291.pcd',
       'n015-2018-10-02-10-50-40+0800__LIDAR_TOP__1538448754047572.pcd',
       'n008-2018-08-28-16-43-51-0400__LIDAR_TOP__1535489299547057.pcd',
       'n015-2018-07-24-11-22-45+0800__LIDAR_TOP__1532402941797653.pcd'],
      dtype='<U62')

Does it need explicit code modification in /CaDDN/tools/train_utils/train_utils.py file to train on nuscenes?

Look forward for any suggestions.
Thank you for your time.

@thom966 thom966 changed the title ImportError: cannot import name 'VoxelGenerator' from 'spconv.utils' Training on nuscenes dataset Apr 24, 2022
@Ng-Alexjc
Copy link

Hello Thom,
may i know how to modify the model config yaml file for nuscenes? i've never done it before. Or where can i get some information about this?
thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants