未验证 提交 8c7126bf 编写于 作者: C cnn 提交者: GitHub

fix doc, set min_subgraph_size of s2anet (#3112)

上级 ecf6a539
...@@ -39,7 +39,7 @@ python deploy/python/infer.py --model_dir=./inference/yolov3_mobilenet_v1_roadsi ...@@ -39,7 +39,7 @@ python deploy/python/infer.py --model_dir=./inference/yolov3_mobilenet_v1_roadsi
| --threshold | No|预测得分的阈值,默认为0.5| | --threshold | No|预测得分的阈值,默认为0.5|
| --output_dir | No|可视化结果保存的根目录,默认为output/| | --output_dir | No|可视化结果保存的根目录,默认为output/|
| --run_benchmark | No| 是否运行benchmark,同时需指定`--image_file``--image_dir` | | --run_benchmark | No| 是否运行benchmark,同时需指定`--image_file``--image_dir` |
| --use_mkldnn | No | CPU预测中是否开启MKLDNN加速 | | --enable_mkldnn | No | CPU预测中是否开启MKLDNN加速 |
| --cpu_threads | No| 设置cpu线程数,默认为1 | | --cpu_threads | No| 设置cpu线程数,默认为1 |
说明: 说明:
......
...@@ -31,7 +31,7 @@ TRT_MIN_SUBGRAPH = { ...@@ -31,7 +31,7 @@ TRT_MIN_SUBGRAPH = {
'SSD': 60, 'SSD': 60,
'RCNN': 40, 'RCNN': 40,
'RetinaNet': 40, 'RetinaNet': 40,
'S2ANet': 40, 'S2ANet': 80,
'EfficientDet': 40, 'EfficientDet': 40,
'Face': 3, 'Face': 3,
'TTFNet': 60, 'TTFNet': 60,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册