Skip to content

Commit 938ef53

Browse files
authored
Improve mmdet3d doc (#1394)
* docs(mmdet3d): add trt version desc * docs(mmdet3d): update
1 parent 0da6059 commit 938ef53

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

docs/en/04-supported-codebases/mmdet3d.md

+6-3
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,14 @@ python tools/deploy.py \
2727

2828
| Model | Task | OnnxRuntime | TensorRT | ncnn | PPLNN | OpenVINO | Model config |
2929
| :----------: | :------------: | :---------: | :------: | :--: | :---: | :------: | :------------------------------------------------------------------------------------: |
30-
| PointPillars | VoxelDetection | Y | Y | N | N | Y | [config](https://github.com/open-mmlab/mmdetection3d/blob/master/configs/pointpillars) |
30+
| PointPillars | VoxelDetection | Y | Y\* | N | N | Y | [config](https://github.com/open-mmlab/mmdetection3d/blob/master/configs/pointpillars) |
3131

32-
## Reminder
32+
1. mmdet3d models on **cu102+TRT8.4** can be visualized normally. For cuda-11 or TRT8.2 users, these issues should be checked
3333

34-
Voxel detection onnx model excludes model.voxelize layer and model post process, and you can use python api to call these func.
34+
- [TRT8.2 assertion `is_tensor`](https://github.com/NVIDIA/TensorRT/issues/1541)
35+
- [TRT8.4 output NaN](https://github.com/NVIDIA/TensorRT/issues/2338)
36+
37+
2. Voxel detection onnx model excludes model.voxelize layer and model post process, and you can use python api to call these func.
3538

3639
Example:
3740

docs/zh_cn/04-supported-codebases/mmdet3d.md

+6-3
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,14 @@ python tools/deploy.py \
2727

2828
| Model | Task | OnnxRuntime | TensorRT | ncnn | PPLNN | OpenVINO | Model config |
2929
| :----------: | :------------: | :---------: | :------: | :--: | :---: | :------: | :------------------------------------------------------------------------------------: |
30-
| PointPillars | VoxelDetection | Y | Y | N | N | Y | [config](https://github.com/open-mmlab/mmdetection3d/blob/master/configs/pointpillars) |
30+
| PointPillars | VoxelDetection | Y | Y\* | N | N | Y | [config](https://github.com/open-mmlab/mmdetection3d/blob/master/configs/pointpillars) |
3131

32-
## 注意事项
32+
1. TensorRT 实测 **cu102+TRT8.4** 可以正常可视化,使用 cuda-11 或 TRT8.2 版本需关注这些 issue
3333

34-
体素检测 onnx 模型不包含 model.voxelize 层和模型后处理,可用 python api 来调这些函数。
34+
- [TRT8.2 assertion `is_tensor`](https://github.com/NVIDIA/TensorRT/issues/1541)
35+
- [TRT8.4 output NaN](https://github.com/NVIDIA/TensorRT/issues/2338)
36+
37+
2. 体素检测 onnx 模型不包含 model.voxelize 层和模型后处理,可用 python api 来调这些函数。
3538

3639
示例:
3740

0 commit comments

Comments
 (0)