Support waymo_info_test.pkl and ImageSets generating #2180
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Motivation
Currently, mmdet3d-dev1.x cannot generate waymo_info_test.pkl for testing or testing_camera_only properly.
This PR is aiming at fixing it.
Modification
create_ImageSets_img_ids()
to generate ImageSets according to data in kitti_format/update_pkl_infos('waymo', out_dir=out_dir, pkl_path=test_path)
to update test info generated inkitti.create_waymo_info_file(...)
timestamp
collecting fromvelodyne_path
collecting.timestamp
is very important link between kitti_format data and tfrecords data, so we still need to collect it even if no lidar data provided. PR More customizable and Faster waymo evaluation #2179 also need timestamps in waymo_infos_*.pkl to generate idx2metainfo.pkldict.get(key)
methods instead ofdict[key]
methods to support updating info files without lidar data or annotations. Now functionupdate_waymo_infos
is able to updatewaymo_infos_test.pkl
or the camera-only one.if range_image_top_pose is None: return
in functionsave_lidar
. In this way, we don't need to judge if the split is camera_only or not.Issue
Currently the easiest way to generate
waymo camera-only test set pkl file
is renamingtesting_3d_camera_only_detection
totesting
, and turn offvelodyne
inWaymoInfoGatherer
. Actually, simply adding an item could only generate cam-only test data in kitti_format but offers no pkl file. How do we plan to fix it?