Skip to content

Commit d045c67

Browse files
committed
Support PyTorch2ONNX and ONNX2TRT for PointPillars
1 parent 57af3f5 commit d045c67

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,12 @@ A Simple PointPillars PyTorch Implenmentation for 3D Lidar(KITTI) Detection. [[Z
66
- Only one detection network (PointPillars) was implemented in this repo, so the code may be more easy to read.
77
- Sincere thanks for the great open-source architectures [mmcv](https://github.com/open-mmlab/mmcv), [mmdet](https://github.com/open-mmlab/mmdetection) and [mmdet3d](https://github.com/open-mmlab/mmdetection3d), which helps me to learn 3D detetion and implement this repo.
88

9+
## News
10+
11+
- **2024-04** Exporting PointPillars to ONNX & TensorRT is supported on branch [feature/deployment](https://github.com/zhulf0804/PointPillars/tree/feature/deployment).
12+
13+
![](./figures/pytorch_trt.png)
14+
915
## mAP on KITTI validation set (Easy, Moderate, Hard)
1016

1117
| Repo | Metric | Overall | Pedestrian | Cyclist | Car |

figures/pytorch_trt.png

126 KB
Loading

0 commit comments

Comments
 (0)