The yolo series model does not get weights information via onnx.model.graph.input
Nobody has claimed this yet.
- 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
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- 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