paddle2onnxPaddlePaddle 到 ONNX 模型轉(zhuǎn)換器
paddle2onnx 是一個 PaddlePaddle 到 ONNX 模型轉(zhuǎn)換器，支持將 PaddlePaddle 框架下產(chǎn)出的模型轉(zhuǎn)化到 ONNX 模型格式。
## CPU版本安裝命令
pip install -f https://paddlepaddle.org.cn/pip/oschina/cpu paddlepaddle
## GPU版本安裝命令
pip install -f https://paddlepaddle.org.cn/pip/oschina/gpu paddlepaddle-gpu
參數(shù)說明
| 參數(shù) | 參數(shù)說明 |
|---|---|
| fluid_model | paddle fluid模型和模型參數(shù)所在目錄 |
| onnx_model | 轉(zhuǎn)化成onnx模型的模型名稱 |
| name_prefix | [可選]某些paddle模型的模型參數(shù)加了前綴，則需要指定模型參數(shù)前綴,例如@HUB_mobilenet_v2_imagenet@conv6_2_expand_bn_scale |
| fluid_model_name | [可選]如果導(dǎo)入的paddle模型不是默認__model__,需要指定模型的名字 |
| fluid_params_name | [可選]如果導(dǎo)入的paddle模型參數(shù)是合并在一個文件里面，需要指定模型參數(shù)文件名 |
| debug | [可選]如果開發(fā)者要對轉(zhuǎn)化的模型進行精度測試，打開此開關(guān) |
| return_variable | [可選]在debug模式中，如果paddle模型返回的結(jié)果是LoDTensor,需要打開此開關(guān) |
| check_task | [可選]在debug模式中，根據(jù)不同配置項選擇不同的執(zhí)行器和數(shù)據(jù)構(gòu)造器 |
| image_path | [可選]在debug模式中，可以選擇加載不同的圖片進行精度驗證 |
評論
圖片
表情