模型转换参数一览表
更新时间：2024-08-22
| 框架 | 网络 | ModelToOnnx 参数 | 盒子类型 | SDK | OnnxToModel 参数 | 备注 |
|---|---|---|---|---|---|---|
| PaddlePaddle | ppyoloe | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | igie_1.0.0 | --input ["image","scale_factor"] --shape [[1,3,640,640],[1,2]] --precision fp16 --algorithm_name ppyoloe | ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| PaddlePaddle | ppyoloe | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | ixrt_4.0.0 | 带nms、静态batch： --algorithm_name ppyoloe --precision fp16 --input_nodenames image --output_nodenames p2o.Mul.224 concat_14.tmp_0 --iou_threshold 0.5 --score_threshold 0.3 --max_output_boxes 1000 --with_nms 带nms、动态batch: --algorithm_name ppyoloe --precision fp16 --input_nodenames image --output_nodenames p2o.Mul.224 concat_14.tmp_0 --iou_threshold 0.5 --score_threshold 0.3 --max_output_boxes 1000 --with_nms --min_shape image:1x3x640x640 --opt_shape image:2x3x640x640 --max_shape image:4x3x640x640 --dynamic_batch |
ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| PaddlePaddle | ppyoloe-tiny | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | igie_1.0.0 | -- | ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| PaddlePaddle | ppyoloe-tiny | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | ixrt_4.0.0 | 带nms、动态batch: --algorithm_name ppyoloe_tiny --precision fp16 --input_nodenames p2o.transpose_0.tmp_0.0 --output_nodenames p2o.Mul.224 concat_16.tmp_0 --iou_threshold 0.5 --score_threshold 0.3 --max_output_boxes 1000 --min_shape p2o.transpose_0.tmp_0.0:1x3x640x640 --opt_shape p2o.transpose_0.tmp_0.0:2x3x640x640 --max_shape p2o.transpose_0.tmp_0.0:4x3x640x640 --with_nms --dynamic_batch |
ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| PaddlePaddle | pplcnet | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | igie_1.0.0 | --input ["x"] --shape [[1,3,256,192]] --precision fp16 --algorithm_name pplcnet | ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| PaddlePaddle | pplcnet | --model_dir ./ --model_filename ruixing_phase2_trash_bin_detection.pdmodel --params_filename ruixing_phase2_trash_bin_detection.pdiparams --save_file model.onnx --enable_dev_version True --opset_version 13 --enable_onnx_checker True | 天数 | ixrt_4.0.0 | --algorithm_name pplcnet --batch_size -1 --input_height 256 --input_width 192 --precision fp16 --min_shape x:1x3x256x192 --opt_shape x:8x3x256x192 --max_shape x:32x3x256x192 | ModelToOnnx参数： --model_filename与--params_filename要和上传的模型文件解压出来的模型结构文件和模型参数文件名字一致 |
| Pytorch | yolov7 | --grid --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 | BM1684 | sophon_3.0.0、 sophon_0.4 |
--outdir=output --target=BM1684 --shapes=[[1,3,640,640]] --net_name=yolov7_fp32_person_640 --opt=2 --dyn=False --cmp=True --enable_profile=True | -- |
| Pytorch | yolov7 | --grid --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 | BM1684X | sophon_3.0.0、 sophon_0.4 |
--outdir=output --target=BM1684X --shapes=[[1,3,640,640]] --net_name=yolov7_fp32_person_640 --opt=2 --dyn=False --cmp=True --enable_profile=True | -- |
| Pytorch | yolov7 | --grid --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 | Rockchip (瑞芯微通用) |
-- | --model_arch=yolov7 --precision=fp --platform=rk3576 | OnnxToModel 参数： --precision 表示精度，可选fp或int --platform 表示目标盒子类型，支持范围包括["rk3576", "rk3568", "rk3566", "rk3588", "rk3562", "rv1103", "rk2118", "rv1106"] |
| Pytorch | yolov7 | --grid --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 | Xavier NX | Jetpack5 | -- | -- |