onnx / onnx/models

unable to replicate yolov4 onnx conversion example

Open
#390 10 comments 2 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?

yolov4

Describe the bug

Attempted to replicate results using the steps described in Conversion.ipynb
The result is a much different graph that gives errors when running with the inference.ipynb
It only generates one output of shape: Output shape: [(1, 33, 84)]

Whereas your onnx model has output shape: Output shape: [(1, 52, 52, 3, 85), (1, 26, 26, 3, 85), (1, 13, 13, 3, 85)]
How did you change the output shape?

Reproduction instructions

System Information

OS Platform and Distribution (Linux Ubuntu 18.04):
ONNX version (1.7.0):
Backend/Runtime version (ONNX Runtime 1.5.2):

Provide a code snippet to reproduce your errors.
run the suggested tutorials as written:
https://github.com/onnx/models/blob/master/vision/object_detection_segmentation/yolov4/dependencies/Conversion.ipynb
https://github.com/onnx/models/blob/master/vision/object_detection_segmentation/yolov4/dependencies/inference.ipynb

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 with dependencies/Conversion.ipynb and dependencies/inference.ipynb, then compare the converted graph with the repository's yolov4 model and its reported output shapes. Reproduce the notebooks using the stated ONNX and ONNX Runtime versions; done means the conversion produces the three expected outputs and the inference notebook runs without the reported errors.

Written by the indexing model from the issue text.

Assessment

Tech stack
jupyter-notebook
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.