votenet pre-trained scannet model doesn't work! KeyError: 'ann_info'

[simba0703]

raceback (most recent call last):
File "demo/pcd_demo.py", line 41, in
main()
File "demo/pcd_demo.py", line 28, in main
result, data = inference_detector(model, args.pcd)
File "/home/user/deeplearning/mmdetection3d/mmdet3d/apis/inference.py", line 102, in inference_detector
data = test_pipeline(data)
File "/home/user/deeplearning/mmdetection/mmdet/datasets/pipelines/compose.py", line 40, in call
data = t(data)
File "/home/user/deeplearning/mmdetection3d/mmdet3d/datasets/pipelines/transforms_3d.py", line 406, in call
assert 'axis_align_matrix' in input_dict['ann_info'].keys(),
KeyError: 'ann_info'

[Wuziyi616]

Thank you for pointing out the bug. I will raise a PR to fix it.
This bug is because we only support VoteNet demo on SUN RGB-D dataset, and ScanNet is slightly different from SUN RGB-D. Maybe you can try playing with the SUN RGB-D VoteNet weight first?