A project demonstrating Lidar related AI solutions, including three GPU accelerated Lidar/camera DL networks (PointPillars, CenterPoint, BEVFusion) and the related libs (cuPCL, 3D SparseConvolution, YUV2RGB, cuOSD,).
Other
1.27k
stars
224
forks
source link
I train centerpoint with kitti, so how can i get rpn_centerhead_sim.onnx and centerpoint.scn.onnx?? #259
I want to use custom data, two ways. First one, transfer kitti type data to nuscenes, I dont find correct way, failed.
Second one, train centerpoint with kitti, I get pth file, but I cant get rpn_centerhead_sim.onnx and centerpoint.scn.onnx with Lidar_AI_Solution/CUDA-CenterPoint/tool/export_neck_head.py and Lidar_AI_Solution/CUDA-CenterPoint/qat/onnx_export/export-scn.py.
Can u give me some suggestion??
I want to use custom data, two ways. First one, transfer kitti type data to nuscenes, I dont find correct way, failed. Second one, train centerpoint with kitti, I get pth file, but I cant get rpn_centerhead_sim.onnx and centerpoint.scn.onnx with Lidar_AI_Solution/CUDA-CenterPoint/tool/export_neck_head.py and Lidar_AI_Solution/CUDA-CenterPoint/qat/onnx_export/export-scn.py. Can u give me some suggestion??