Commit cfe42b5c authored by Julien's avatar Julien
Browse files

added a helper function to get directly nier from string

parent 4c2cb7d4
......@@ -482,5 +482,12 @@ let produce_cfg (g:Graph_proto.t) =
to_edge)
shared_name_preds;
(* else (); *)
cfg
cfg
let nier_from_onnx input_file =
let model = parse_model_from_file input_file in
let graph = match model.Model_proto.graph with
| Some g -> g
| None -> raise (ParseError
"No graph in ONNX input file!")
in produce_cfg graph
......@@ -33,3 +33,6 @@ val parse_model_from_file : string -> Onnx_piqi.Model_proto.t
*)
val produce_cfg : Onnx_piqi.Graph_proto.t -> NCFG.NierCFG.t
val nier_from_onnx : string -> NCFG.NierCFG.t
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment