Add nuScenes data processing into codebase - #6
Conversation
|
Some images compared: Although nuscene has different coordinates (left-hand) compared to others, ... add more analysis here later. 一些注意点(This will be translated once I done notes here)
So, at the end, I think I finished the draft script on nuscene data extraction! I will soon update the script here while the training might need a double check as the gt frames number is really small compared to the other two (10%, so the deflow might be similar to seflow fig. 4?) Some frames and scene count:
Updates: Checked with seflow trained on nuscene, it works well. and also the nus gt training on deltaflow works well also. [I will update more results table on nuscene once deltaflow is accepted.) More result will be added in the paper. |
ce99b49 to
8ff2678
Compare
8ff2678 to
3d53932
Compare
* update related README also. * add NusMap to argoverse 2 for consistent evaluation afterward.
|
More nuScene visualization here (for id: 21, scene_id: scene-0757, timestamp: 1535657118649480):
Extraction code already updated. The training dataloader will be merged through the DeltaFlow project. Training Result from DeltaFlow:
I will check the policy about nuScene to see if we can upload pretrained weights for users, etc. |
* will nus training need update dataloader, it will inovled by DeltaFlow merge there.
ec56be7 to
d16adf5
Compare
* conf(nus): update ground segmentation in nuscenes. * conf(nus): update ground segmentation in nuscenes. * feat(data): core extraction py file for nuscenes. * Add NusMap to Argoverse 2 for consistent evaluation afterward. * style(model): update a basemodel to avoid repeating the same fn, etc. * will nus training need to update the dataloader, it will involved by DeltaFlow merge there. * docs: fix typo on dataprocess README. tested successfully on all fn. * Add flow check in nus from supervised training later. --------- Co-authored-by: EdwardLeeLPZ <EdwardLeeLPZ@users.noreply.github.com>



This pull request introduces major improvements to nuScenes dataset support, including a new extraction pipeline, configuration options, documentation updates, and utility functions. The changes enable standardized 10Hz resampling, ground truth scene flow generation, and robust handling of annotated frames, making the nuScenes processing consistent with other datasets and easier to use.
nuScenes dataset extraction and processing:
dataprocess/extract_nus.pyscript for nuScenes dataset extraction, including 10Hz resampling, GT annotation handling, ground segmentation, and per-object velocity-based scene flow computation. The script supports parallel processing and robust output file management. [1] [2]conf/others/nuscenes.tomlfor nuScenes ground segmentation and point cloud partitioning parameters.check_h5py_file_existsand nuScenes-to-Argoverse category mapping (NusNamMap) indataprocess/misc_data.pyto ensure output consistency and correct class mapping. [1] [2]Documentation updates:
README.mdanddataprocess/README.mdto reflect nuScenes dataset support, including extraction instructions, frame rate explanation, and dataset statistics. [1] [2] [3] [4]Model base class:
BaseModelclass tosrc/models/basic/__init__.pyfor standardized model checkpoint loading, improving code reuse and maintainability.[Old message]
As requested by many users, this pull request integrates nuScenes data processing scripts into the codebase.