Failed to load YOLOv3 and Faster RCNN model

[zhangpiu]

I'm trying to load YOLOv3 and Faster RCNN model, but hit errors.

julia> using Flux, ONNX

julia> ONNX.load_model("yolov3.onnx")
ERROR: KeyError: key :Squeeze not found
Stacktrace:
 [1] getindex(::Dict{Symbol,Any}, ::Symbol) at ./dict.jl:478
 [2] _graph(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:0
 [3] graph(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:92
 [4] code(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:100
 [5] write_julia_file(::String) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/convert.jl:235
 [6] load_model(::String) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/convert.jl:250
 [7] top-level scope at none:0

julia> using Flux, ONNX

julia> ONNX.load_model("faster_rcnn_R_50_FPN_1x.onnx")
ERROR: KeyError: key :Gather not found
Stacktrace:
 [1] getindex(::Dict{Symbol,Any}, ::Symbol) at ./dict.jl:478
 [2] _graph(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:0
 [3] graph(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:92
 [4] code(::ONNX.Types.Graph) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/graph/graph.jl:100
 [5] write_julia_file(::String) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/convert.jl:235
 [6] load_model(::String) at /Users/yuanye/.julia/packages/ONNX/iz0dL/src/convert.jl:250
 [7] top-level scope at none:0

[gabrielpreviato]

I'm having the same issue with Julia v1.5.2 and ONNX.jl v0.1.1