File tree 2 files changed +54
-0
lines changed
2 files changed +54
-0
lines changed Original file line number Diff line number Diff line change @@ -202,3 +202,30 @@ And the output look like this:
202
202
| Max | 1.689 | 591.983 |
203
203
+--------+------------+---------+
204
204
```
205
+
206
+ ## visualize
207
+
208
+ This tool can be used to visualize model inference results in different backend.
209
+
210
+ ### Usage
211
+
212
+ ``` bash
213
+ python tools/visualize.py \
214
+ --deploy-cfg {DEPLOY_CFG} \
215
+ --model-cfg {MODEL_CFG} \
216
+ --deploy-path {DEPLOY_PATH} \
217
+ --test-img {TEST_IMGS} \
218
+ --checkpoint {CHECKPOINTS} \
219
+ --save-dir {SAVE_DIR} \
220
+ --device {DEVICE}
221
+ ```
222
+
223
+ ### Description of all arguments
224
+
225
+ - ` deploy-cfg ` : The path of the deploy config file in MMDeploy codebase.
226
+ - ` model-cfg ` : The path of model config file in OpenMMLab codebase.
227
+ - ` deploy-path ` : The path of the model to be tested, if the backend contains multiple files, you can use it multiple times.
228
+ - ` test-img ` : The path of the images to be tested, you can use it multiple times.
229
+ - ` checkpoint ` : The path of the checkpoint to be tested, if it is used, the result will be cancated to right part.
230
+ - ` save-dir ` : The path to save the visualization results, if it not specified, it will be set to '.'.
231
+ - ` device ` : The device type. If not specified, it will be set to ` cpu ` .
Original file line number Diff line number Diff line change @@ -202,3 +202,30 @@ python tools/profiler.py \
202
202
| Max | 1.689 | 591.983 |
203
203
+--------+------------+---------+
204
204
```
205
+
206
+ ## visualize
207
+
208
+ 这个工具可以用来可视化不同推理引擎的推理结果。
209
+
210
+ ### 用法
211
+
212
+ ``` bash
213
+ python tools/visualize.py \
214
+ --deploy-cfg {DEPLOY_CFG} \
215
+ --model-cfg {MODEL_CFG} \
216
+ --deploy-path {DEPLOY_PATH} \
217
+ --test-img {TEST_IMGS} \
218
+ --checkpoint {CHECKPOINTS} \
219
+ --save-dir {SAVE_DIR} \
220
+ --device {DEVICE}
221
+ ```
222
+
223
+ ### 参数说明
224
+
225
+ - ` deploy-cfg ` : 输入的配置文件路径
226
+ - ` model-cfg ` : 输入的模型配置文件路径
227
+ - ` deploy-path ` : 测试的模型文件路径,如果部分模型含有多个文件,请多次使用该参数
228
+ - ` test-img ` : 测试的图片路径,可以多次使用测试测试多张图片
229
+ - ` checkpoint ` : PyTorch的权重文件,如果使用这个参数,推理的结果会被拼接到图像的右侧
230
+ - ` save-dir ` : 保存可视化结果,如果没有指定则会被设为当前目录
231
+ - ` device ` : 运行的设备类型,如果没有指定则会默认设置为` cpu `
You can’t perform that action at this time.
0 commit comments