ldkong1205 / LaserMix

[CVPR 2023 Highlight] LaserMix for Semi-Supervised LiDAR Semantic Segmentation
https://ldkong.com/LaserMix
Apache License 2.0
272 stars 17 forks source link

Cylender3D Experiments #7

Closed christy0621 closed 11 months ago

christy0621 commented 11 months ago

It is fine if you don't want to release the semi-supervised code. But is there any plan to release the supervised code of Cylender3D in the Nuscenes dataset?


Edit:

I tried your setup (with a lower voxel grid size) based on the original Cylinder3D code https://github.com/xinge008/Cylinder3D, but the results are far lower than the results reported in your paper.

ldkong1205 commented 11 months ago

Hi @christy0621, as mentioned previously (https://github.com/ldkong1205/LaserMix/issues/6), we are working on the journal extension of this work. The complete code should be out in the next several weeks.

For the results of Cylinder3D:

christy0621 commented 11 months ago

1). I used the voxel size (240, 180, 20) (following your paper), which is different from the original implementation. I got 73.0 for the full label, and 66.4 for the 50% label, based on fully supervised learning.

2). whether your implementation of Cylinder3D model based on Torchsparse or Spconv? My implementation is based on the code you released, which is torchsparse.

ldkong1205 commented 11 months ago

Hi @christy0621, that's it. The current code for implementing Cylinder3D uses a torchsparse backend from SPVCNN, which is faster (in terms of both training and inference speed) but the performance becomes lower.

The results reported in the paper are from a spconv 1.2.1 Cylinder3D, the same as the original implementation (https://github.com/xinge008/Cylinder3D). You should be able to reproduce the results by switching the backend back to spconv.

Our next update will include support for both backends, leaving room for you to choose between accuracy and efficiency. Please stay tuned for this update!