onnx / onnx/models

The yolo series model does not get weights information via onnx.model.graph.input

Open
#479 7 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

bug
Dominant language
Jupyter Notebook
Stars
9.8k
Forks
1.6k
PR merge metrics
No merged PRs in 30d

Description

Bug Report

Which model does this pertain to?

https://github.com/onnx/models/blob/master/vision/object_detection_segmentation/yolov3/model/yolov3-10.onnx
https://github.com/onnx/models/blob/master/vision/object_detection_segmentation/yolov4/model/yolov4.onnx

Describe the bug

Describe the problem with the model and the results you are seeing.

Reproduction instructions

OS Platform and Distribution (e.g. Linux Ubuntu 16.04):
Linux VM-1-159-ubuntu 4.15.0-136-generic #140-Ubuntu SMP Thu Jan 28 05:20:47 UTC 2021 x86_64 x86_64 x86_64 GNU/Linux

ONNX version (e.g. 1.6):
Name: onnx
Version: 1.10.1
Summary: Open Neural Network Exchange
Home-page: https://github.com/onnx/onnx
Author: ONNX
Author-email: onnx-technical-discuss@lists.lfai.foundation
License: Apache License v2.0
Location: /usr/local/anaconda3/envs/ubuntu/lib/python3.8/site-packages
Requires: numpy, typing-extensions, protobuf, six
Required-by:

Provide a code snippet to reproduce your errors.

import onnx
model = onnx.load('yolov3-10.onnx')
print(model.graph.input)
Output

[name: "input_1"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_param: "unk__576"
}
dim {
dim_value: 3
}
dim {
dim_param: "unk__577"
}
dim {
dim_param: "unk__578"
}
}
}
}
, name: "image_shape"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_param: "unk__579"
}
dim {
dim_value: 2
}
}
}
}
]

Notes

The input array in GraphProto contains not only the node that we generally understand as the image input, but also all the weights in the model

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by loading the linked yolov3-10.onnx and yolov4.onnx files with the provided onnx.load snippet and inspecting each model's graph.input. Compare the reported inputs with the issue's expectation about weights, then define the corrected model metadata and verify both models expose the intended information.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.