Skip to content

Fix TorchScript JSON string key bug - #9015

Merged
glenn-jocher merged 2 commits into
masterfrom
glenn-jocher-patch-1
Aug 18, 2022
Merged

Fix TorchScript JSON string key bug#9015
glenn-jocher merged 2 commits into
masterfrom
glenn-jocher-patch-1

Conversation

@glenn-jocher

@glenn-jocher glenn-jocher commented Aug 18, 2022 •
edited by UltralyticsAssistant
Loading

Copy link
Copy Markdown
Member

Resolves #9011

Signed-off-by: Glenn Jocher glenn.jocher@ultralytics.com

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improved metadata handling in YOLOv5 TorchScript model loading.

📊 Key Changes

  • Enhanced JSON parsing to convert string keys to integers when loading model metadata, if they are digits.
  • Maintained the previous structure of loading model metadata from an 'extra_files' dictionary.

🎯 Purpose & Impact

  • Purpose: To ensure that metadata such as stride values and class names can be accurately retrieved with correct data types, which is important for downstream tasks such as data preprocessing and inference.
  • Impact: Users can expect more robust model loading, especially when using TorchScript models in various deployment scenarios. This update reduces the risk of type-related errors when the model metadata is utilized during inference. 🚀

@glenn-jocher
Fix TorchScript JSON string key bug
c167b41 
Resolves #9011

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher self-assigned this Aug 18, 2022
@glenn-jocher
glenn-jocher merged commit 529aafd into master Aug 18, 2022
@glenn-jocher
glenn-jocher deleted the glenn-jocher-patch-1 branch August 18, 2022 09:50
@glenn-jocher glenn-jocher mentioned this pull request Aug 18, 2022
Closed
1 task done
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
@glenn-jocher @pre-commit-ci Clay Januhowski
Fix TorchScript JSON string key bug (ultralytics#9015)
5fbdb3c 
* Fix TorchScript JSON string key bug

Resolves ultralytics#9011

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
@hojland hojland mentioned this pull request Oct 17, 2022
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

@glenn-jocher glenn-jocher

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Fail to run torchscript after exporting

1 participant

@glenn-jocher