Skip to content
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

YOLOv5 OpenVINO PyTorch Hub inference fix #7826

Merged
merged 1 commit into from
May 15, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 15, 2022

Resolves bug report #7817

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved model checkpoint path resolution in hubconf.py.

πŸ“Š Key Changes

  • Modified the checkpoint path assignment logic to handle directories properly.

🎯 Purpose & Impact

  • 🎯 Purpose: Ensure that when specifying a model name, the code does not mistakenly add a .pt suffix to a directory, preventing potential loading errors.
  • πŸ’₯ Impact: This improvement will lead to more robust model loading, especially when using custom directories or unconventional naming schemes. Users will experience fewer errors and more predictable behavior when loading models.

@glenn-jocher glenn-jocher linked an issue May 15, 2022 that may be closed by this pull request
1 task
@glenn-jocher glenn-jocher merged commit 1e112ce into master May 15, 2022
@glenn-jocher glenn-jocher deleted the fix/openvino branch May 15, 2022 22:56
@glenn-jocher glenn-jocher self-assigned this May 15, 2022
tdhooghe pushed a commit to tdhooghe/yolov5 that referenced this pull request Jun 10, 2022
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

What to do with output tensor (OpenVINO Inference)
1 participant