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

d #13255

Merged
merged 7 commits into from
Sep 16, 2024
Merged

d #13255

merged 7 commits into from
Sep 16, 2024

Conversation

yothinsaengs
Copy link
Contributor

@yothinsaengs yothinsaengs commented Aug 9, 2024

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

This PR introduces the CBAM (Convolutional Block Attention Module) to YOLOv5 for improved attention mechanism.

πŸ“Š Key Changes

  • Added CAM (Channel Attention Module) and SAM (Spatial Attention Module) classes.
  • Implemented CBAM combining CAM and SAM.
  • Updated yolo.py to support CBAM.
  • Introduced a new model configuration yolov5s_CBAM.yaml.

🎯 Purpose & Impact

  • Enhanced Feature Extraction: CBAM improves the model's attention to relevant image features, which can increase detection accuracy.
  • Better Performance: Enhanced focus on important features may lead to improved object detection performance, especially in complex scenes.
  • New Model Option: Offers users a new configuration to experiment with advanced attention mechanisms.

Copy link
Contributor

github-actions bot commented Aug 9, 2024


Thank you for your submission, we really appreciate it. Like many open-source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution. You can sign the CLA by just posting a Pull Request Comment same as the below format.


I have read the CLA Document and I sign the CLA


You can retrigger this bot by commenting recheck in this Pull Request. Posted by the CLA Assistant Lite bot.

Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

πŸ‘‹ Hello @yothinsaengs, thank you for submitting a YOLOv5 πŸš€ PR! To allow your work to be integrated as seamlessly as possible, we advise you to:

  • βœ… Verify your PR is up-to-date with ultralytics/yolov5 master branch. If your PR is behind you can update your code by clicking the 'Update branch' button or by running git pull and git merge master locally.
  • βœ… Verify all YOLOv5 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

@glenn-jocher glenn-jocher merged commit e2d6b25 into ultralytics:master Sep 16, 2024
6 of 8 checks passed
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.

2 participants