onnxruntime-web
Version:
A Javascript library for running ONNX models on browsers
62 lines • 2.47 kB
JavaScript
;
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
Object.defineProperty(exports, "__esModule", { value: true });
exports.Model = void 0;
const flatbuffers_1 = require("flatbuffers");
const graph_1 = require("./graph");
const ort_generated_1 = require("./ort-schema/flatbuffers/ort-generated");
const onnx_1 = require("./ort-schema/protobuf/onnx");
const util_1 = require("./util");
var ortFbs = ort_generated_1.onnxruntime.experimental.fbs;
class Model {
// empty model
constructor() { }
load(buf, graphInitializer, isOrtFormat) {
if (!isOrtFormat) {
// isOrtFormat === false || isOrtFormat === undefined
try {
this.loadFromOnnxFormat(buf, graphInitializer);
return;
}
catch (e) {
if (isOrtFormat !== undefined) {
throw e;
}
}
}
this.loadFromOrtFormat(buf, graphInitializer);
}
loadFromOnnxFormat(buf, graphInitializer) {
const modelProto = onnx_1.onnx.ModelProto.decode(buf);
const irVersion = util_1.LongUtil.longToNumber(modelProto.irVersion);
if (irVersion < 3) {
throw new Error('only support ONNX model with IR_VERSION>=3');
}
this._opsets =
modelProto.opsetImport.map(i => ({ domain: i.domain, version: util_1.LongUtil.longToNumber(i.version) }));
this._graph = graph_1.Graph.from(modelProto.graph, graphInitializer);
}
loadFromOrtFormat(buf, graphInitializer) {
const fb = new flatbuffers_1.flatbuffers.ByteBuffer(buf);
const ortModel = ortFbs.InferenceSession.getRootAsInferenceSession(fb).model();
const irVersion = util_1.LongUtil.longToNumber(ortModel.irVersion());
if (irVersion < 3) {
throw new Error('only support ONNX model with IR_VERSION>=3');
}
this._opsets = [];
for (let i = 0; i < ortModel.opsetImportLength(); i++) {
const opsetId = ortModel.opsetImport(i);
this._opsets.push({ domain: opsetId?.domain(), version: util_1.LongUtil.longToNumber(opsetId.version()) });
}
this._graph = graph_1.Graph.from(ortModel.graph(), graphInitializer);
}
get graph() {
return this._graph;
}
get opsets() {
return this._opsets;
}
}
exports.Model = Model;
//# sourceMappingURL=model.js.map