p2o 0.1.1

A PaddlePaddle New IR (PIR) to ONNX model converter.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use clap::Parser;

use p2o::cli::Cli;
use p2o::converter::Converter;

fn main() -> anyhow::Result<()> {
    env_logger::init();
    let cli = Cli::parse();
    log::info!("=== PaddlePaddle to ONNX Converter ===");
    log::info!("Target Opset: {}", cli.opset);

    let mut converter = Converter::new();
    converter.set_target_opset(cli.opset);
    converter.set_strict(cli.strict);
    converter.load_paddle_model(&cli.model_json)?;
    converter.load_paddle_weights(&cli.model_pdiparams)?;
    converter.export_onnx(&cli.output_onnx, cli.opset)?;
    Ok(())
}