Skip to content

Fix numpy to torch cls streaming bug#9112

Merged
glenn-jocher merged 2 commits intomasterfrom
glenn-jocher-patch-1
Aug 23, 2022
Merged

Fix numpy to torch cls streaming bug#9112
glenn-jocher merged 2 commits intomasterfrom
glenn-jocher-patch-1

Conversation

@glenn-jocher
Copy link
Copy Markdown
Member

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

Resolves #9111

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

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improved image tensor conversion in YOLOv5 classification predict script.

📊 Key Changes

  • Added an explicit import statement for torch.
  • Modified the image tensor preparation to ensure it is converted to a PyTorch tensor before being sent to the device.

🎯 Purpose & Impact

  • These changes ensure compatibility and proper tensor handling across different versions of PyTorch or various input formats.
  • Users can expect more reliable and consistent behavior when using the classification model on diverse image inputs, leading to smoother user experiences and potentially better classification performance. 🚀

@glenn-jocher
Fix numpy to torch cls streaming bug
99a620f 
Resolves #9111

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher self-assigned this Aug 23, 2022
@glenn-jocher glenn-jocher merged commit e6f54c5 into master Aug 23, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch August 23, 2022 21:54
@glenn-jocher glenn-jocher mentioned this pull request Aug 23, 2022
Closed
2 tasks
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
@glenn-jocher @pre-commit-ci
Fix numpy to torch cls streaming bug (ultralytics#9112)
890da37 
* Fix numpy to torch cls streaming bug

Resolves ultralytics#9111

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.

Classification from webcam

1 participant

@glenn-jocher