-
-
Notifications
You must be signed in to change notification settings - Fork 16.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update export.py #5471
Update export.py #5471
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
👋 Hello @nanmi, thank you for submitting a 🚀 PR! To allow your work to be integrated as seamlessly as possible, we advise you to:
- ✅ Verify your PR is up-to-date with upstream/master. If your PR is behind upstream/master an automatic GitHub actions rebase may be attempted by including the /rebase command in a comment body, or by running the following code, replacing 'feature' with the name of your local branch:
git remote add upstream https://github.com/ultralytics/yolov5.git
git fetch upstream
git checkout feature # <----- replace 'feature' with local branch name
git merge upstream/master
git push -u origin -f
- ✅ Verify all Continuous Integration (CI) checks are passing.
- ✅ Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." -Bruce Lee
@nanmi hi there. What is the bug you are trying to fix with this? For ONNX dynamic models the correct export command is: python export.py --include onnx --dynamic |
@glenn-jocher Hi, when I fine tune the model and save onnx directly, it is easy to generate output the processing after the detection head, if I export the pre training model onnx separately, there will be no error 😀 |
@nanmi I'm sorry I'm not understanding. The Detect layer onnx_dynamic attribute is already set correctly here: Lines 293 to 295 in ac2c49a
If you are seeing an error in YOLOv5 that this PR is fixing, please raise a bug report first and then link this PR to it as a solution. As is I don't understand the problem this PR is fixing. |
/rebase |
* fix export onnx bug * Update export.py * Update export.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update yolo.py Co-authored-by: Glenn Jocher <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
🛠️ PR Summary
Made with ❤️ by Ultralytics Actions
🌟 Summary
This PR appears to be either an empty or placeholder PR, with no code changes included.
📊 Key Changes
🎯 Purpose & Impact