search

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
527 stars 155 forks source link

fix onnx exporter bug #52

Open PeterJaq opened 2 years ago

PeterJaq commented 2 years ago

This PR is related to https://github.com/NVIDIA-AI-IOT/CUDA-PointPillars/issues/50, This commit fix the dummy_input['batch_size'] is int not supported as JIT inputs/outputs.