Skip to content

Commit

Permalink
[Frontend-Superior-MaskRCNN]
Browse files Browse the repository at this point in the history
Change-Id: I9bb56292c7cd3ab9becf476f1298bcb5f3ae4aa7
  • Loading branch information
seriphdog authored and charlesxzb committed Sep 18, 2024
1 parent ac4c024 commit fdae946
Show file tree
Hide file tree
Showing 58 changed files with 5,762 additions and 19 deletions.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
14 changes: 12 additions & 2 deletions docs/developer_manual/source_en/03_user_interface.rst
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,9 @@ Used to convert various neural network models into MLIR files, the supported par
* - model_def
- Y
- Model definition file (e.g., '.onnx', '.tflite' or '.prototxt' files)
* - model_extern
- N
- Extra multi model definition files (currently mainly used for MaskRCNN). None by default. separate by ','
* - model_data
- N
- Specify the model weight file, required when it is caffe model (corresponding to the '.caffemodel' file)
Expand Down Expand Up @@ -254,11 +257,15 @@ Used to convert various neural network models into MLIR files, the supported par
* - preprocess_list
- N
- choose which input need preprocess, like:'1,3' means input 1&3 need preprocess, default all inputs

* - path_yaml
- N
- the path for one single yaml file (currently mainly used for MaskRCNN)
* - enable_maskrcnn
- N
- if enable MaskRCNN transformation

After converting to an mlir file, a ``${model_name}_in_f32.npz`` file will be generated, which is the input file for the subsequent models.


run_calibration.py
~~~~~~~~~~~~~~~~~~~~~~~~~

Expand Down Expand Up @@ -598,6 +605,9 @@ Convert the mlir file into the corresponding model, the parameters are as follow
* - matmul_perchannel
- N
- if matmul is quantized in per-channel mode, for BM1684X and BM1688, the performance may be decreased if enable
* - enable_maskrcnn
- N
- if enable comparison for MaskRCNN.

model_runner.py
~~~~~~~~~~~~~~~~
Expand Down
Loading

0 comments on commit fdae946

Please sign in to comment.