/pointpillars_pytorch_trt

Primary LanguageC++Apache License 2.0Apache-2.0

A Deployment Implementation For PointPillars

This repository is a deployment implementation for pointpillars. It has two parts: a PyTorch implementation for ONNX export and a TensorRT implementation for deployment.

PyTorch

We train and export the model based on OpenPCDet.

Go to pytorch for more details.

TensorRT

The deployment implementation is a modified version of TensorRT and tiny-tensorrt.

Go to trt for more details.

Contributors

Zhongyuan Qiu, Hao Liu, Chenchen Zhang, Bo Wen.

Novauto 超星未来

Novauto.png

Acknowledgement

The code is developed based on OpenPCDet , TensorRT and tiny-tensorrt.

License

This project is licensed under the Apache license 2.0 License - see the LICENSE file for details.