Skip to content

how to use mini dataset #80

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

Open
lightning0016 opened this issue Sep 23, 2024 · 3 comments
Open

how to use mini dataset #80

lightning0016 opened this issue Sep 23, 2024 · 3 comments

Comments

@lightning0016
Copy link

when i run :
python tools/data_converter/vad_nuscenes_converter.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag vad_nuscenes --version v1.0-mini --canbus ./data
run scueess
when run:
CUDA_VISIBLE_DEVICES=0 python tools/test.py projects/configs/VAD/VAD_tiny_e2e.py /home/wxl/VAD/ckpts/VAD_tiny.pth --launcher none --eval bbox --tmpdir tmp
i has error:
No such file or directory: 'data/nuscenes/maps/expansion/boston-seaport.json'

how to use mini dataset

@lightning0016
Copy link
Author

oh,i know not down map

@C-Moger
Copy link

C-Moger commented Oct 13, 2024

How do I split trainval and test if I want to use a mini data set

@Tornadosky
Copy link

This is what I found to be working for me:

python tools/data_converter/vad_nuscenes_converter.py nuscenes --root-path ../data/nuscenes --out-dir ../data/nuscenes --extra-tag vad_mini --version v1.0-mini --canbus ../data/nuscenes

This will create .pkl files in the nuscenes directory. extra-tag is just a prefix your files will have.
In the config file, e.g. VAD_tiny_e2e.py you have to replace data_root with your path to nuscenes and also change the ann_file (that you created above) in the data dict, e.g. ann_file=data_root + 'vad_mini_infos_temporal_val.pkl'. What I also did in commented out the map_ann_file in the train, val, and test.

You also have to have CanBus extension and a MapExtension for this case.

Then you are free to do the training with:

python tools/train.py projects/configs/VAD/VAD_tiny_e2e.py --launcher none --work-dir ../tmp/VAD

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

3 participants