add axis_align_matrix (#600)

open-mmlab · May 31, 2021 · a629806 · a629806
1 parent e65772d
commit a629806
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmdet3d/apis/inference.py b/mmdet3d/apis/inference.py
@@ -89,6 +89,8 @@ def inference_detector(model, pcd):
  pts_filename=pcd,
  box_type_3d=box_type_3d,
  box_mode_3d=box_mode_3d,
+ # for ScanNet demo we need axis_align_matrix
+ ann_info=dict(axis_align_matrix=np.eye(4)),
  sweeps=[],
  # set timestamp = 0
  timestamp=[0],