DerrickXuNu/v2x-vit

bug in early_fusion_dataset.py

memberRE opened this issue · 0 comments

Hi, runsheng,
In early_fusion_dataset.py line 99 to 110, seems like ‘unique_indices’ should also be updated.
unique_indices = unique_indices[:object_bbx_center_valid.shape[0]]
Or it may cause error in BasePostprocessor.generate_gt_bbx() line 91

https://github.com/DerrickXuNu/v2x-vit/blob/main/v2xvit/data_utils/datasets/early_fusion_dataset.py#L99