[Fix] Fix numpy ValueError in update_infos_to_v2.py #2162
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
Fix numpy value assignment error. I was updating an info.pkl generated from an older mmdet3d of v1.0.0rc2, via command
python tools/dataset_converters/update_infos_to_v2.py --dataset nuscenes --pkl-path ./data/nus_rc2/nuscenes_infos_val.pkl --out-dir ./data/nus_v2
I encountered this error:
File "tools/dataset_converters/update_infos_to_v2.py", line 336, in update_nuscenes_infos lidar2sensor[:3, 3] = -1 * np.matmul(rot.T, trans.reshape(3, 1))
ValueError: could not broadcast input array from shape (3,1) into shape (3,)
Modification
In update_infos_to_v2.py, I simply changed all
lidar2sensor[:3, 3] = -1 * np.matmul(rot.T, trans.reshape(3, 1))
tolidar2sensor[:3, 3:4] = -1 * np.matmul(rot.T, trans.reshape(3, 1))