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

Update models/hub/*.yaml files for v6.0n release #5540

Merged
merged 3 commits into from
Nov 6, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 6, 2021

Updates all extra YOLOv5 models in the models/hub directory to v6.0 release architecture.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Upgraded YOLOv5 configurations to v6.0, introducing model structure enhancements and feature integrations.

πŸ“Š Key Changes

  • Updated backbone and head definitions across various YOLOv5 model configurations (BiFPN, FPN, P2, P6, P7, PANet, Ghost, and Transformer versions).
  • Replaced Focus layers with Conv layers with updated arguments.
  • Adjusted the number of C3 layers in backbone configurations.
  • Swapped SPP (Spatial Pyramid Pooling) layers with the more efficient SPPF layers.
  • Marked concatenations specifically within the BiFPN configuration.
  • Updated the Transformer (C3TR) layer configuration.
  • Added functionality to test all YOLOv5 models via a command-line argument.

🎯 Purpose & Impact

  • The transition from Focus to Conv layers is expected to standardize the initial convolution process across various models.
  • Reducing the C3 layers might optimize computational efficiency without significantly affecting model accuracy.
  • The switch to SPPF should streamline spatial pyramid pooling while preserving feature diversity with reduced computational overhead.
  • Explicitly marking concatenation steps enhances code readability and could help in debugging the BiFPN model.
  • Refining the Transformer model aligns with updates in transformer technology and potentially boosts performance.
  • The added test argument (--test) simplifies the process of verification for all model configurations, ensuring robustness before deployment.
  • These changes aim to improve the models' performance, resource utilization, and maintainability, which benefits users with more efficient and effective object detection tools. πŸš€πŸ§ 

πŸ› οΈ Overall, the PR represents a stride towards evolving and optimizing the YOLOv5 architecture. Users might benefit from faster, sleeker, and possibly more accurate object detection in applications.

@glenn-jocher glenn-jocher self-assigned this Nov 6, 2021
@glenn-jocher glenn-jocher linked an issue Nov 6, 2021 that may be closed by this pull request
1 task
@glenn-jocher glenn-jocher merged commit fa2344c into master Nov 6, 2021
@glenn-jocher glenn-jocher deleted the update/model_yamls branch November 6, 2021 14:07
@glenn-jocher glenn-jocher mentioned this pull request Nov 6, 2021
1 task
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Update model yamls for v6.0

* Add python models/yolo.py --test

* Ghost fix
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.

small objects
1 participant