Use VoxelNext to training custom dataset about voxelnext

[yukang2017]

Hi Yibo,

Thanks for your interest in our work. After carefully reading your issue, I suggest that there are two choices to try.

1. Disable IoU branch
The IoU branch is just for the Waymo dataset. I think it might be useless for others.
Set
IOU_BRANCH: Falsein the config file.

And remove
'iou': {'out_channels': 1, 'num_conv': 2},in the HEAD_DICT.

2. Try VoxelNeXt backbone with other head.
After reading your comment, I think what your really like is the VoxelResBackBone8xVoxelNeXt backbone network. You can use it with other detector head, like Voxel R-CNN head. What you additionally need to do is just to convert the sparse out of VoxelResBackBone8xVoxelNeXt to dense, following the common VoxelResBackBone8x.

In addition, I am very happy for more discussion via WeChat. I will contact you latter.

Regards,
Yukang Chen

[hoangduyloc]

Hi @luoxiaoliaolan, @yukang2017

I have problem in training custom dataset. My dataset has only Lidar (x, y, z), and it works normally with Second, Pointpillar, PVRCNN config, but with VoxelNext It wont work. I tried and have the same errors as you. Would you guide me abit to fix it.

Thank you both!