[doc] refine maskrcnn

Change-Id: I2fc9d76c550e150be8266eb5bcbceaeffaab094e
sophgo · Sep 25, 2024 · 975c7f2 · 975c7f2
1 parent 951e859
commit 975c7f2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/docs/developer_manual/source_zh/19_maskrcnn.rst b/docs/developer_manual/source_zh/19_maskrcnn.rst
@@ -125,7 +125,7 @@ MaskRCNN 大算子框架需要 5 个输入:
 
     $ tool_maskrcnn.py \
       --path_yaml              ./regression/dataset/MaskRCNN/CONFIG_MaskRCNN.yaml \
-      --path_input_image       ./regression/dataset/MaskRCNN/Superior_IMG_BackBone.npz \
+      --path_input_image       Superior_IMG_BackBone.npz \
       --basic_max_shape_inverse 1216,800 \
       --basic_scalar_factor     1.8734375,1.8735363 \
       --debug
@@ -170,12 +170,12 @@ IO_MAP 指南
  * **数据类型**:仅支持 fp32 或 int32.
  * **连接**: 每个输入找出其来源的上层模块(可能不是上一个相邻模块), 和上层模块相应输出的操作序数.
 
-请注意, -1 表示完整 MaskRCNN 的输入, 而 -2 表示完整模型的输出, Top
+请注意, -1 表示完整 MaskRCNN 的输入, 而 -2 表示完整模型的输出
 
 **[-1] Top_In**
 
 .. list-table::
-   :widths: 5 15 50 5
+   :widths: 15 25 30 10
    :header-rows: 1
 
    * - 输入编号

diff --git a/regression/dataset/MaskRCNN/CONFIG_MaskRCNN.yaml b/regression/dataset/MaskRCNN/CONFIG_MaskRCNN.yaml
@@ -143,3 +143,9 @@ TOPK_ONNX_NMS_1st: 250
 TOPK_ONNX_NMS_2nd: 250
 TPU1686_TOPK_FACTOR: 3
 W_RPN_DYN_MAX: 304
+maskrcnn_model_transform_extern_params:
+  list_encoded_block: 1,0,0,1,0,0,1
+  list_name_PPLop: ppl::RPN_get_bboxes,ppl::Bbox_Pooler,ppl::get_bboxes_B,ppl::Mask_Pooler
+  maskrcnn_io_map: '{(0,0):(-1,0),(1,0):(0,5),(1,1):(0,6),(1,2):(0,7),(1,3):(0,8),(1,4):(0,9),(1,5):(0,10),(1,6):(0,11),(1,7):(0,12),(1,8):(0,13),(1,9):(0,14),(1,10):(-1,1),(2,0):(0,0),(2,1):(0,1),(2,2):(0,2),(2,3):(0,3),(2,4):(1,0),(3,0):(2,0),(4,0):(2,1),(4,1):(3,1),(4,2):(3,0),(4,3):(-1,2),(4,4):(-1,3),(5,0):(0,0),(5,1):(0,1),(5,2):(0,2),(5,3):(0,3),(5,4):(4,0),(5,5):(4,1),(5,6):(-1,4),(6,0):(5,0),(-2,0):(4,0),(-2,1):(4,1),(-2,2):(6,0)}'
+  maskrcnn_num_output: '3'
+  nums_PPLOp_Inputs: 11,5,5,7