NVIDIA-AI-IOT / CUDA-PointPillars

A project demonstrating how to use CUDA-PointPillars to deal with cloud points data from lidar.
Apache License 2.0
502 stars 148 forks source link

how to convert .pt into onnx for pointpillars of mmdetection3d? #74

Open sylivahf opened 1 year ago

sylivahf commented 1 year ago

@NVIDIA-AI-IOT thank you for your works,this is very helpful for me. but there are some questions in mmdet3d conversion:i can't directly convert the model of mmdet3d, and i have no specific idea of revision ,because there are differences in the definition and use of pointpillars in mmdet3d and pcdet。 --------------mmdet3d-pointpillars-------------------- 0 backbone.blocks.0.0.weight 1 backbone.blocks.0.1.weight 2 backbone.blocks.0.1.bias 3 backbone.blocks.0.3.weight 4 backbone.blocks.0.4.weight 5 backbone.blocks.0.4.bias 6 backbone.blocks.0.6.weight 7 backbone.blocks.0.7.weight 8 backbone.blocks.0.7.bias 9 backbone.blocks.0.9.weight 10 backbone.blocks.0.10.weight 11 backbone.blocks.0.10.bias 12 backbone.blocks.1.0.weight 13 backbone.blocks.1.1.weight 14 backbone.blocks.1.1.bias 15 backbone.blocks.1.3.weight 16 backbone.blocks.1.4.weight 17 backbone.blocks.1.4.bias 18 backbone.blocks.1.6.weight 19 backbone.blocks.1.7.weight 20 backbone.blocks.1.7.bias 21 backbone.blocks.1.9.weight 22 backbone.blocks.1.10.weight 23 backbone.blocks.1.10.bias 24 backbone.blocks.1.12.weight 25 backbone.blocks.1.13.weight 26 backbone.blocks.1.13.bias 27 backbone.blocks.1.15.weight 28 backbone.blocks.1.16.weight 29 backbone.blocks.1.16.bias 30 backbone.blocks.2.0.weight 31 backbone.blocks.2.1.weight 32 backbone.blocks.2.1.bias 33 backbone.blocks.2.3.weight 34 backbone.blocks.2.4.weight 35 backbone.blocks.2.4.bias 36 backbone.blocks.2.6.weight 37 backbone.blocks.2.7.weight 38 backbone.blocks.2.7.bias 39 backbone.blocks.2.9.weight 40 backbone.blocks.2.10.weight 41 backbone.blocks.2.10.bias 42 backbone.blocks.2.12.weight 43 backbone.blocks.2.13.weight 44 backbone.blocks.2.13.bias 45 backbone.blocks.2.15.weight 46 backbone.blocks.2.16.weight 47 backbone.blocks.2.16.bias 48 neck.deblocks.0.0.weight 49 neck.deblocks.0.1.weight 50 neck.deblocks.0.1.bias 51 neck.deblocks.1.0.weight 52 neck.deblocks.1.1.weight 53 neck.deblocks.1.1.bias 54 neck.deblocks.2.0.weight 55 neck.deblocks.2.1.weight 56 neck.deblocks.2.1.bias 57 bbox_head.conv_cls.weight 58 bbox_head.conv_cls.bias 59 bbox_head.conv_reg.weight 60 bbox_head.conv_reg.bias 61 bbox_head.conv_dir_cls.weight 62 bbox_head.conv_dir_cls.bias 63 voxel_encoder.pfn_layers.0.norm.weight 64 voxel_encoder.pfn_layers.0.norm.bias 65 voxel_encoder.pfn_layers.0.linear.weight

--------------pcdet-pointpillars----------------------------------- 0 vfe.pfn_layers.0.linear.weight 1 vfe.pfn_layers.0.norm.weight 2 vfe.pfn_layers.0.norm.bias 3 backbone_2d.blocks.0.1.weight 4 backbone_2d.blocks.0.2.weight 5 backbone_2d.blocks.0.2.bias 6 backbone_2d.blocks.0.4.weight 7 backbone_2d.blocks.0.5.weight 8 backbone_2d.blocks.0.5.bias 9 backbone_2d.blocks.0.7.weight 10 backbone_2d.blocks.0.8.weight 11 backbone_2d.blocks.0.8.bias 12 backbone_2d.blocks.0.10.weight 13 backbone_2d.blocks.0.11.weight 14 backbone_2d.blocks.0.11.bias 15 backbone_2d.blocks.1.1.weight 16 backbone_2d.blocks.1.2.weight 17 backbone_2d.blocks.1.2.bias 18 backbone_2d.blocks.1.4.weight 19 backbone_2d.blocks.1.5.weight 20 backbone_2d.blocks.1.5.bias 21 backbone_2d.blocks.1.7.weight 22 backbone_2d.blocks.1.8.weight 23 backbone_2d.blocks.1.8.bias 24 backbone_2d.blocks.1.10.weight 25 backbone_2d.blocks.1.11.weight 26 backbone_2d.blocks.1.11.bias 27 backbone_2d.blocks.1.13.weight 28 backbone_2d.blocks.1.14.weight 29 backbone_2d.blocks.1.14.bias 30 backbone_2d.blocks.1.16.weight 31 backbone_2d.blocks.1.17.weight 32 backbone_2d.blocks.1.17.bias 33 backbone_2d.blocks.2.1.weight 34 backbone_2d.blocks.2.2.weight 35 backbone_2d.blocks.2.2.bias 36 backbone_2d.blocks.2.4.weight 37 backbone_2d.blocks.2.5.weight 38 backbone_2d.blocks.2.5.bias 39 backbone_2d.blocks.2.7.weight 40 backbone_2d.blocks.2.8.weight 41 backbone_2d.blocks.2.8.bias 42 backbone_2d.blocks.2.10.weight 43 backbone_2d.blocks.2.11.weight 44 backbone_2d.blocks.2.11.bias 45 backbone_2d.blocks.2.13.weight 46 backbone_2d.blocks.2.14.weight 47 backbone_2d.blocks.2.14.bias 48 backbone_2d.blocks.2.16.weight 49 backbone_2d.blocks.2.17.weight 50 backbone_2d.blocks.2.17.bias 51 backbone_2d.deblocks.0.0.weight 52 backbone_2d.deblocks.0.1.weight 53 backbone_2d.deblocks.0.1.bias 54 backbone_2d.deblocks.1.0.weight 55 backbone_2d.deblocks.1.1.weight 56 backbone_2d.deblocks.1.1.bias 57 backbone_2d.deblocks.2.0.weight 58 backbone_2d.deblocks.2.1.weight 59 backbone_2d.deblocks.2.1.bias 60 dense_head.conv_cls.weight 61 dense_head.conv_cls.bias 62 dense_head.conv_box.weight 63 dense_head.conv_box.bias 64 dense_head.conv_dir_cls.weight 65 dense_head.conv_dir_cls.bias