ONNXEngine | DOCSAID

📄️ EngineConfig

EngineConfig

getonnxinput_infos(model: str | Path | onnx.ModelProto) -> dict[str, dict[str, Any]]

getonnxmetadata(onnx_path: str | Path) -> dict[str, Any]

getonnxoutput_infos(model: str | Path | onnx.ModelProto) -> dict[str, dict[str, Any]]

makeonnxdynamicaxes(modelfpath str | Path, inputdims dict[str, dict[int, str]], opsetversion: int | None = None) -> None

ONNXEngine(modelpath int = 0, backend Mapping[str, Any] | None = None, provideroption EngineConfig | None = None)

parsemetadatafromonnx(onnxpath: str | Path) -> dict[str, Any]

writemetadataintoonnx(onnxpath str | Path, dropold_meta Any) -> None