Skip to content

GitLab

Unverified Commit 11d8554c authored by Zaida Zhou's avatar Zaida Zhou Committed by GitHub
Browse files

[Docs] Fix mdformat version to support python3.6 (#2014) 

* [Docs] Refine documentation

* fix mdformat version to support python3.6

* remove ruby dependency from CI

* fix invalid links
parent 834d5978
Show whitespace changes
Inline Side-by-side
Showing with 209 additions and 213 deletions
.github/workflows/lint.yml
View file @ 11d8554c
...@@ -17,10 +17,6 @@ jobs: ...@@ -17,10 +17,6 @@ jobs:
python-version: 3.7 python-version: 3.7
- name: Install pre-commit hook - name: Install pre-commit hook
run: | run: |
# markdownlint requires ruby >= 2.7
sudo apt-add-repository ppa:brightbox/ruby-ng -y
sudo apt-get update
sudo apt-get install -y ruby2.7
pip install pre-commit pip install pre-commit
pre-commit install pre-commit install
- name: Linting - name: Linting
......
.pre-commit-config.yaml
View file @ 11d8554c
...@@ -30,12 +30,12 @@ repos: ...@@ -30,12 +30,12 @@ repos:
hooks: hooks:
- id: codespell - id: codespell
- repo: https://github.com/executablebooks/mdformat - repo: https://github.com/executablebooks/mdformat
rev: 0.7.14 rev: 0.7.9
hooks: hooks:
- id: mdformat - id: mdformat
args: ["--number"] args: ["--number"]
additional_dependencies: additional_dependencies:
- mdformat-gfm - mdformat-openmmlab
- mdformat_frontmatter - mdformat_frontmatter
- linkify-it-py - linkify-it-py
- repo: https://github.com/myint/docformatter - repo: https://github.com/myint/docformatter
......
TERMINOLOGY.md
View file @ 11d8554c
...@@ -5,7 +5,7 @@ This document is used as a reference for English-Chinese terminology translation ...@@ -5,7 +5,7 @@ This document is used as a reference for English-Chinese terminology translation
该文档用作中英文翻译对照参考。 该文档用作中英文翻译对照参考。
| English | 中文 | | English | 中文 |
| :---------------: | :----: | | :---------------: | :----------: |
| annotation | 标注 | | annotation | 标注 |
| backbone | 主干网络 | | backbone | 主干网络 |
| benchmark | 基准测试 | | benchmark | 基准测试 |
......
docs/en/deployment/onnxruntime_op.md
View file @ 11d8554c
...@@ -107,7 +107,7 @@ Take custom operator `soft_nms` for example. ...@@ -107,7 +107,7 @@ Take custom operator `soft_nms` for example.
2. Add source `soft_nms.cpp` to ONNX Runtime source directory `mmcv/ops/csrc/onnxruntime/cpu/` 2. Add source `soft_nms.cpp` to ONNX Runtime source directory `mmcv/ops/csrc/onnxruntime/cpu/`
3. Register `soft_nms` operator in [onnxruntime_register.cpp](../../mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp) 3. Register `soft_nms` operator in [onnxruntime_register.cpp](../../../mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp)
```c++ ```c++
#include "soft_nms.h" #include "soft_nms.h"
...@@ -133,4 +133,4 @@ Take custom operator `soft_nms` for example. ...@@ -133,4 +133,4 @@ Take custom operator `soft_nms` for example.
### References ### References
- [How to export Pytorch model with custom op to ONNX and run it in ONNX Runtime](https://github.com/onnx/tutorials/blob/master/PyTorchCustomOperator/README.md) - [How to export Pytorch model with custom op to ONNX and run it in ONNX Runtime](https://github.com/onnx/tutorials/blob/master/PyTorchCustomOperator/README.md)
- [How to add a custom operator/kernel in ONNX Runtime](https://github.com/microsoft/onnxruntime/blob/master/docs/AddingCustomOp.md) - [How to add a custom operator/kernel in ONNX Runtime](https://onnxruntime.ai/docs/reference/operators/add-custom-op.html)
docs/zh_cn/deployment/onnxruntime_custom_ops.md
View file @ 11d8554c
...@@ -2,55 +2,55 @@ ...@@ -2,55 +2,55 @@
<!-- TOC --> <!-- TOC -->
- [ONNX Runtime自定义算子](#onnx-runtime%E8%87%AA%E5%AE%9A%E4%B9%89%E7%AE%97%E5%AD%90) - [ONNX Runtime自定义算子](#onnx-runtime自定义算子)
- [SoftNMS](#softnms) - [SoftNMS](#softnms)
- [描述](#%E6%8F%8F%E8%BF%B0) - [描述](#描述)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0) - [模型参数](#模型参数)
- [输入](#%E8%BE%93%E5%85%A5) - [输入](#输入)
- [输出](#%E8%BE%93%E5%87%BA) - [输出](#输出)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F) - [类型约束](#类型约束)
- [RoIAlign](#roialign) - [RoIAlign](#roialign)
- [描述](#%E6%8F%8F%E8%BF%B0-1) - [描述](#描述-1)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-1) - [模型参数](#模型参数-1)
- [输入](#%E8%BE%93%E5%85%A5-1) - [输入](#输入-1)
- [输出](#%E8%BE%93%E5%87%BA-1) - [输出](#输出-1)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-1) - [类型约束](#类型约束-1)
- [NMS](#nms) - [NMS](#nms)
- [描述](#%E6%8F%8F%E8%BF%B0-2) - [描述](#描述-2)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-2) - [模型参数](#模型参数-2)
- [输入](#%E8%BE%93%E5%85%A5-2) - [输入](#输入-2)
- [输出](#%E8%BE%93%E5%87%BA-2) - [输出](#输出-2)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-2) - [类型约束](#类型约束-2)
- [grid_sampler](#grid_sampler) - [grid_sampler](#grid_sampler)
- [描述](#%E6%8F%8F%E8%BF%B0-3) - [描述](#描述-3)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-3) - [模型参数](#模型参数-3)
- [输入](#%E8%BE%93%E5%85%A5-3) - [输入](#输入-3)
- [输出](#%E8%BE%93%E5%87%BA-3) - [输出](#输出-3)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-3) - [类型约束](#类型约束-3)
- [CornerPool](#cornerpool) - [CornerPool](#cornerpool)
- [描述](#%E6%8F%8F%E8%BF%B0-4) - [描述](#描述-4)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-4) - [模型参数](#模型参数-4)
- [输入](#%E8%BE%93%E5%85%A5-4) - [输入](#输入-4)
- [输出](#%E8%BE%93%E5%87%BA-4) - [输出](#输出-4)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-4) - [类型约束](#类型约束-4)
- [cummax](#cummax) - [cummax](#cummax)
- [描述](#%E6%8F%8F%E8%BF%B0-5) - [描述](#描述-5)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-5) - [模型参数](#模型参数-5)
- [输入](#%E8%BE%93%E5%85%A5-5) - [输入](#输入-5)
- [输出](#%E8%BE%93%E5%87%BA-5) - [输出](#输出-5)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-5) - [类型约束](#类型约束-5)
- [cummin](#cummin) - [cummin](#cummin)
- [描述](#%E6%8F%8F%E8%BF%B0-6) - [描述](#描述-6)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-6) - [模型参数](#模型参数-6)
- [输入](#%E8%BE%93%E5%85%A5-6) - [输入](#输入-6)
- [输出](#%E8%BE%93%E5%87%BA-6) - [输出](#输出-6)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-6) - [类型约束](#类型约束-6)
- [MMCVModulatedDeformConv2d](#mmcvmodulateddeformconv2d) - [MMCVModulatedDeformConv2d](#mmcvmodulateddeformconv2d)
- [描述](#%E6%8F%8F%E8%BF%B0-7) - [描述](#描述-7)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-7) - [模型参数](#模型参数-7)
- [输入](#%E8%BE%93%E5%85%A5-7) - [输入](#输入-7)
- [输出](#%E8%BE%93%E5%87%BA-7) - [输出](#输出-7)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-7) - [类型约束](#类型约束-7)
<!-- TOC --> <!-- TOC -->
...@@ -63,7 +63,7 @@ ...@@ -63,7 +63,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | --------------- | -------------------------------------------------- | | ------- | --------------- | ------------------------------------------------------- |
| `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 | | `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 |
| `float` | `sigma` | 高斯方法的超参数 | | `float` | `sigma` | 高斯方法的超参数 |
| `float` | `min_score` | NMS的score阈值 | | `float` | `min_score` | NMS的score阈值 |
...@@ -101,7 +101,7 @@ ...@@ -101,7 +101,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | ---------------- | ----------------------------------- | | ------- | ---------------- | ------------------------------------------------------- |
| `int` | `output_height` | roi特征的输出高度 | | `int` | `output_height` | roi特征的输出高度 |
| `int` | `output_width` | roi特征的输出宽度 | | `int` | `output_width` | roi特征的输出宽度 |
| `float` | `spatial_scale` | 输入检测框的缩放系数 | | `float` | `spatial_scale` | 输入检测框的缩放系数 |
...@@ -138,7 +138,7 @@ ...@@ -138,7 +138,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | --------------- | ----------------------------------- | | ------- | --------------- | ------------------------------------------------------- |
| `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 | | `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 |
| `int` | `offset` | 用来计算候选框的宽高(x2 - x1 + offset)。可选值0或1 | | `int` | `offset` | 用来计算候选框的宽高(x2 - x1 + offset)。可选值0或1 |
...@@ -171,7 +171,7 @@ ...@@ -171,7 +171,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | -------------------- | ------------------------------------------------------------------------------------------------ | | ----- | -------------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------- |
| `int` | `interpolation_mode` | 计算输出使用的插值模式。(0: `bilinear` , 1: `nearest`) | | `int` | `interpolation_mode` | 计算输出使用的插值模式。(0: `bilinear` , 1: `nearest`) |
| `int` | `padding_mode` | 边缘填充模式。(0: `zeros`, 1: `border`, 2: `reflection`) | | `int` | `padding_mode` | 边缘填充模式。(0: `zeros`, 1: `border`, 2: `reflection`) |
| `int` | `align_corners` | 如果`align_corners=1`，则极值(`-1`和`1`)会被当做输入边缘像素的中心点。如果`align_corners=0`，则它们会被看做是边缘像素的边缘点,减小分辨率对采样的影响 | | `int` | `align_corners` | 如果`align_corners=1`，则极值(`-1`和`1`)会被当做输入边缘像素的中心点。如果`align_corners=0`，则它们会被看做是边缘像素的边缘点,减小分辨率对采样的影响 |
...@@ -205,7 +205,7 @@ ...@@ -205,7 +205,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | ------ | --------------------------------------------------- | | ----- | ------ | -------------------------------------------------------- |
| `int` | `mode` | 池化模式。(0: `top`, 1: `bottom`, 2: `left`, 3: `right`) | | `int` | `mode` | 池化模式。(0: `top`, 1: `bottom`, 2: `left`, 3: `right`) |
#### 输入 #### 输入
...@@ -235,7 +235,7 @@ ...@@ -235,7 +235,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | ----- | --------- | | ----- | ------ | ------------------ |
| `int` | `dim` | 进行累计计算的维度 | | `int` | `dim` | 进行累计计算的维度 |
#### 输入 #### 输入
...@@ -267,7 +267,7 @@ ...@@ -267,7 +267,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | ----- | --------- | | ----- | ------ | ------------------ |
| `int` | `dim` | 进行累计计算的维度 | | `int` | `dim` | 进行累计计算的维度 |
#### 输入 #### 输入
...@@ -299,7 +299,7 @@ ...@@ -299,7 +299,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| -------------- | ------------------- | -------------------------------------- | | -------------- | ------------------- | ------------------------------------------------------------- |
| `list of ints` | `stride` | 卷积的步长 (sH, sW) | | `list of ints` | `stride` | 卷积的步长 (sH, sW) |
| `list of ints` | `padding` | 输入特征填充大小 (padH, padW) | | `list of ints` | `padding` | 输入特征填充大小 (padH, padW) |
| `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) | | `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) |
......
docs/zh_cn/deployment/onnxruntime_op.md
View file @ 11d8554c
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
### MMCV已支持的算子 ### MMCV已支持的算子
| 算子 | CPU | GPU | MMCV版本 | | 算子 | CPU | GPU | MMCV版本 |
| :------------------------------------------------------------------------------: | :-: | :-: | :----: | | :------------------------------------------------------------------------------: | :-: | :-: | :------: |
| [SoftNMS](onnxruntime_custom_ops.md#softnms) | Y | N | 1.2.3 | | [SoftNMS](onnxruntime_custom_ops.md#softnms) | Y | N | 1.2.3 |
| [RoIAlign](onnxruntime_custom_ops.md#roialign) | Y | N | 1.2.5 | | [RoIAlign](onnxruntime_custom_ops.md#roialign) | Y | N | 1.2.5 |
| [NMS](onnxruntime_custom_ops.md#nms) | Y | N | 1.2.7 | | [NMS](onnxruntime_custom_ops.md#nms) | Y | N | 1.2.7 |
...@@ -100,7 +100,7 @@ onnx_results = sess.run(None, {'input' : input_data}) ...@@ -100,7 +100,7 @@ onnx_results = sess.run(None, {'input' : input_data})
2. 在ONNX Runtime源码目录`mmcv/ops/csrc/onnxruntime/cpu/`下添加算子实现`soft_nms.cpp` 2. 在ONNX Runtime源码目录`mmcv/ops/csrc/onnxruntime/cpu/`下添加算子实现`soft_nms.cpp`
3. 在[onnxruntime_register.cpp](../../mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp)中注册实现的算子`soft_nms` 3. 在[onnxruntime_register.cpp](../../../mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp)中注册实现的算子`soft_nms`
```c++ ```c++
#include "soft_nms.h" #include "soft_nms.h"
...@@ -126,4 +126,4 @@ onnx_results = sess.run(None, {'input' : input_data}) ...@@ -126,4 +126,4 @@ onnx_results = sess.run(None, {'input' : input_data})
### 引用 ### 引用
- [How to export Pytorch model with custom op to ONNX and run it in ONNX Runtime](https://github.com/onnx/tutorials/blob/master/PyTorchCustomOperator/README.md) - [How to export Pytorch model with custom op to ONNX and run it in ONNX Runtime](https://github.com/onnx/tutorials/blob/master/PyTorchCustomOperator/README.md)
- [How to add a custom operator/kernel in ONNX Runtime](https://github.com/microsoft/onnxruntime/blob/master/docs/AddingCustomOp.md) - [How to add a custom operator/kernel in ONNX Runtime](https://onnxruntime.ai/docs/reference/operators/add-custom-op.html)
docs/zh_cn/deployment/tensorrt_custom_ops.md
View file @ 11d8554c
...@@ -2,61 +2,61 @@ ...@@ -2,61 +2,61 @@
<!-- TOC --> <!-- TOC -->
- [TensorRT自定义算子](#tensorrt%E8%87%AA%E5%AE%9A%E4%B9%89%E7%AE%97%E5%AD%90) - [TensorRT自定义算子](#tensorrt自定义算子)
- [MMCVRoIAlign](#mmcvroialign) - [MMCVRoIAlign](#mmcvroialign)
- [描述](#%E6%8F%8F%E8%BF%B0) - [描述](#描述)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0) - [模型参数](#模型参数)
- [输入](#%E8%BE%93%E5%85%A5) - [输入](#输入)
- [输出](#%E8%BE%93%E5%87%BA) - [输出](#输出)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F) - [类型约束](#类型约束)
- [ScatterND](#scatternd) - [ScatterND](#scatternd)
- [描述](#%E6%8F%8F%E8%BF%B0-1) - [描述](#描述-1)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-1) - [模型参数](#模型参数-1)
- [输入](#%E8%BE%93%E5%85%A5-1) - [输入](#输入-1)
- [输出](#%E8%BE%93%E5%87%BA-1) - [输出](#输出-1)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-1) - [类型约束](#类型约束-1)
- [NonMaxSuppression](#nonmaxsuppression) - [NonMaxSuppression](#nonmaxsuppression)
- [描述](#%E6%8F%8F%E8%BF%B0-2) - [描述](#描述-2)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-2) - [模型参数](#模型参数-2)
- [输入](#%E8%BE%93%E5%85%A5-2) - [输入](#输入-2)
- [输出](#%E8%BE%93%E5%87%BA-2) - [输出](#输出-2)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-2) - [类型约束](#类型约束-2)
- [MMCVDeformConv2d](#mmcvdeformconv2d) - [MMCVDeformConv2d](#mmcvdeformconv2d)
- [描述](#%E6%8F%8F%E8%BF%B0-3) - [描述](#描述-3)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-3) - [模型参数](#模型参数-3)
- [输入](#%E8%BE%93%E5%85%A5-3) - [输入](#输入-3)
- [输出](#%E8%BE%93%E5%87%BA-3) - [输出](#输出-3)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-3) - [类型约束](#类型约束-3)
- [grid_sampler](#grid_sampler) - [grid_sampler](#grid_sampler)
- [描述](#%E6%8F%8F%E8%BF%B0-4) - [描述](#描述-4)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-4) - [模型参数](#模型参数-4)
- [输入](#%E8%BE%93%E5%85%A5-4) - [输入](#输入-4)
- [输出](#%E8%BE%93%E5%87%BA-4) - [输出](#输出-4)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-4) - [类型约束](#类型约束-4)
- [cummax](#cummax) - [cummax](#cummax)
- [描述](#%E6%8F%8F%E8%BF%B0-5) - [描述](#描述-5)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-5) - [模型参数](#模型参数-5)
- [输入](#%E8%BE%93%E5%85%A5-5) - [输入](#输入-5)
- [输出](#%E8%BE%93%E5%87%BA-5) - [输出](#输出-5)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-5) - [类型约束](#类型约束-5)
- [cummin](#cummin) - [cummin](#cummin)
- [描述](#%E6%8F%8F%E8%BF%B0-6) - [描述](#描述-6)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-6) - [模型参数](#模型参数-6)
- [输入](#%E8%BE%93%E5%85%A5-6) - [输入](#输入-6)
- [输出](#%E8%BE%93%E5%87%BA-6) - [输出](#输出-6)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-6) - [类型约束](#类型约束-6)
- [MMCVInstanceNormalization](#mmcvinstancenormalization) - [MMCVInstanceNormalization](#mmcvinstancenormalization)
- [描述](#%E6%8F%8F%E8%BF%B0-7) - [描述](#描述-7)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-7) - [模型参数](#模型参数-7)
- [输入](#%E8%BE%93%E5%85%A5-7) - [输入](#输入-7)
- [输出](#%E8%BE%93%E5%87%BA-7) - [输出](#输出-7)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-7) - [类型约束](#类型约束-7)
- [MMCVModulatedDeformConv2d](#mmcvmodulateddeformconv2d) - [MMCVModulatedDeformConv2d](#mmcvmodulateddeformconv2d)
- [描述](#%E6%8F%8F%E8%BF%B0-8) - [描述](#描述-8)
- [模型参数](#%E6%A8%A1%E5%9E%8B%E5%8F%82%E6%95%B0-8) - [模型参数](#模型参数-8)
- [输入](#%E8%BE%93%E5%85%A5-8) - [输入](#输入-8)
- [输出](#%E8%BE%93%E5%87%BA-8) - [输出](#输出-8)
- [类型约束](#%E7%B1%BB%E5%9E%8B%E7%BA%A6%E6%9D%9F-8) - [类型约束](#类型约束-8)
<!-- TOC --> <!-- TOC -->
...@@ -69,7 +69,7 @@ ...@@ -69,7 +69,7 @@
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | ---------------- | ----------------------------------- | | ------- | ---------------- | ------------------------------------------------------- |
| `int` | `output_height` | roi特征的输出高度 | | `int` | `output_height` | roi特征的输出高度 |
| `int` | `output_width` | roi特征的输出宽度 | | `int` | `output_width` | roi特征的输出宽度 |
| `float` | `spatial_scale` | 输入检测框的缩放系数 | | `float` | `spatial_scale` | 输入检测框的缩放系数 |
...@@ -148,7 +148,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -148,7 +148,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | ---------------------------- | ----------------------------------------------------------------------------- | | ------- | ---------------------------- | -------------------------------------------------------------------------------------------- |
| `int` | `center_point_box` | 0 - 候选框的格式为\[y1, x1, y2, x2\]， 1-候选框的格式为\[x_center, y_center, width, height\] | | `int` | `center_point_box` | 0 - 候选框的格式为\[y1, x1, y2, x2\]， 1-候选框的格式为\[x_center, y_center, width, height\] |
| `int` | `max_output_boxes_per_class` | 每一类最大的输出检测框个数。默认为0，输出检测框个数等于输入候选框数 | | `int` | `max_output_boxes_per_class` | 每一类最大的输出检测框个数。默认为0，输出检测框个数等于输入候选框数 |
| `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 | | `float` | `iou_threshold` | 用来判断候选框重合度的阈值，取值范围\[0, 1\]。默认值为0 |
...@@ -186,7 +186,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -186,7 +186,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| -------------- | ------------------ | ------------------------------------------------------- | | -------------- | ------------------ | --------------------------------------------------------------------------------------------- |
| `list of ints` | `stride` | 卷积的步长 (sH, sW) | | `list of ints` | `stride` | 卷积的步长 (sH, sW) |
| `list of ints` | `padding` | 输入特征填充大小 (padH, padW) | | `list of ints` | `padding` | 输入特征填充大小 (padH, padW) |
| `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) | | `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) |
...@@ -225,7 +225,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -225,7 +225,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | -------------------- | ------------------------------------------------------------------------------------------------ | | ----- | -------------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------- |
| `int` | `interpolation_mode` | 计算输出使用的插值模式。(0: `bilinear` , 1: `nearest`) | | `int` | `interpolation_mode` | 计算输出使用的插值模式。(0: `bilinear` , 1: `nearest`) |
| `int` | `padding_mode` | 边缘填充模式。(0: `zeros`, 1: `border`, 2: `reflection`) | | `int` | `padding_mode` | 边缘填充模式。(0: `zeros`, 1: `border`, 2: `reflection`) |
| `int` | `align_corners` | 如果`align_corners=1`，则极值(`-1`和`1`)会被当做输入边缘像素的中心点。如果`align_corners=0`，则它们会被看做是边缘像素的边缘点,减小分辨率对采样的影响 | | `int` | `align_corners` | 如果`align_corners=1`，则极值(`-1`和`1`)会被当做输入边缘像素的中心点。如果`align_corners=0`，则它们会被看做是边缘像素的边缘点,减小分辨率对采样的影响 |
...@@ -259,7 +259,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -259,7 +259,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | ----- | --------- | | ----- | ------ | ------------------ |
| `int` | `dim` | 进行累计计算的维度 | | `int` | `dim` | 进行累计计算的维度 |
#### 输入 #### 输入
...@@ -291,7 +291,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -291,7 +291,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ----- | ----- | --------- | | ----- | ------ | ------------------ |
| `int` | `dim` | 进行累计计算的维度 | | `int` | `dim` | 进行累计计算的维度 |
#### 输入 #### 输入
...@@ -323,7 +323,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -323,7 +323,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| ------- | --------- | ----------------- | | ------- | --------- | ---------------------------- |
| `float` | `epsilon` | 用来避免除0错误。默认为1e-05 | | `float` | `epsilon` | 用来避免除0错误。默认为1e-05 |
#### 输入 #### 输入
...@@ -357,7 +357,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i ...@@ -357,7 +357,7 @@ ScatterND接收三个输入，分别为秩为r >= 1的`data`，秩为q >= 1的`i
#### 模型参数 #### 模型参数
| 类型 | 参数名 | 描述 | | 类型 | 参数名 | 描述 |
| -------------- | ------------------- | -------------------------------------- | | -------------- | ------------------- | ------------------------------------------------------------- |
| `list of ints` | `stride` | 卷积的步长 (sH, sW) | | `list of ints` | `stride` | 卷积的步长 (sH, sW) |
| `list of ints` | `padding` | 输入特征填充大小 (padH, padW) | | `list of ints` | `padding` | 输入特征填充大小 (padH, padW) |
| `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) | | `list of ints` | `dilation` | 卷积核各元素间隔 (dH, dW) |
......
docs/zh_cn/deployment/tensorrt_plugin.md
View file @ 11d8554c
...@@ -25,7 +25,7 @@ ...@@ -25,7 +25,7 @@
### MMCV中的TensorRT插件列表 ### MMCV中的TensorRT插件列表
| ONNX算子 | TensorRT插件 | MMCV版本 | | ONNX算子 | TensorRT插件 | MMCV版本 |
| :-----------------------: | :-----------------------------------------------------------------------------: | :----: | | :-----------------------: | :-----------------------------------------------------------------------------: | :------: |
| MMCVRoiAlign | [MMCVRoiAlign](./tensorrt_custom_ops.md#mmcvroialign) | 1.2.6 | | MMCVRoiAlign | [MMCVRoiAlign](./tensorrt_custom_ops.md#mmcvroialign) | 1.2.6 |
| ScatterND | [ScatterND](./tensorrt_custom_ops.md#scatternd) | 1.2.6 | | ScatterND | [ScatterND](./tensorrt_custom_ops.md#scatternd) | 1.2.6 |
| NonMaxSuppression | [NonMaxSuppression](./tensorrt_custom_ops.md#nonmaxsuppression) | 1.3.0 | | NonMaxSuppression | [NonMaxSuppression](./tensorrt_custom_ops.md#nonmaxsuppression) | 1.3.0 |
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment