Pointcept / PointTransformerV3

[CVPR'24 Oral] Official repository of Point Transformer V3 (PTv3)
MIT License
583 stars 30 forks source link

RuntimeError: The size of tensor a (8192) must match the size of tensor b (3) at non-singleton dimension 1 #44

Open SFHhhh opened 1 month ago

SFHhhh commented 1 month ago

my input data :

data_dict = { "feat": feats, # (4, 2048, 6) "coord": coords, # (4, 2048, 3)
"grid_size": 0.05, # 体素尺度0.05米 "offset": offsets, # (4,) tensor([2048,4096,6144,8192]) }

but in default.py 21 lines code = batch << depth * 3 | code RuntimeError: The size of tensor a (8192) must match the size of tensor b (3) at non-singleton dimension 1,Why does this error occur?

Gofinge commented 1 month ago

Hi, the data structure is not correct (e.g. [4, 2048, 6] -> [8192, 6]). Please refer readme here (https://github.com/Pointcept/Pointcept?tab=readme-ov-file#offset)