search

NVIDIA-AI-IOT / Lidar_AI_Solution

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.19k stars 204 forks source link

support of custom dataset #43

Closed shuyuan-wang closed 1 year ago

shuyuan-wang commented 1 year ago

Hi: Does your sparse convolution support inference with input shape N*4? When I'm trying to inference, it says

image

hopef commented 1 year ago

Sorry, the libspconv currently only supports Nx5/Nx8/Nx16 input. We will add further support as needed and release it in the future.

shuyuan-wang commented 1 year ago

Does spatial shape have any limit?

hopef commented 1 year ago

There are no limitations. It just requires more effort.

hopef commented 1 year ago

Now you can update libspconv.so to be able to support the Nx4 input.

wangxu5 commented 1 year ago

Can you please update libsponv.so to be able to support the N*3 input？Thanks~ @hopef

hopef commented 1 year ago

Ok, I will add it