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

New: Ultralytics YOLO-Human #12702

Open
wants to merge 145 commits into
base: main
Choose a base branch
from
Open

New: Ultralytics YOLO-Human #12702

wants to merge 145 commits into from

Conversation

Laughing-q
Copy link
Member

@Laughing-q Laughing-q commented May 15, 2024

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Introducing new YOLOHuman model for human attribute detection! πŸš€

πŸ“Š Key Changes

  • Added YOLOHuman class as part of the model imports.
  • Introduced a new YAML configuration for the YOLOv8 human detection model.
  • Implemented additional augmentations to include human attribute data handling.
  • Established a new dataset class, HumanDataset, for loading and processing human-related datasets.
  • Included Human object in results to encapsulate detected human attributes.
  • Enriched model __init__.py to include YOLOHuman.
  • Formulated HumanPredictor, HumanTrainer, and HumanValidator under the new YOLO human module for prediction, training, and validation.

🎯 Purpose & Impact

  • Enhances Model Catalog: Expands Ultralytics' model offerings to include human-specific attribute detection.
  • Improves Dataset Handling: Offers streamlined process for datasets involving human features.
  • Facilitates Human-centric Applications: Paves the way for more sophisticated applications such as demographic analysis, security enhancements, and personalized customer experiences.

@glenn-jocher
Copy link
Member

@Laughing-q @Burhan-Q I attempted to de-conflict this with YOLOv10 in bba1152, I saw a mismatch here with m.reg_max and m.reg_max - 1 in v10, so went with the v10 version for the de-conflict, not sure if I might have broken something here in the training or if the Human models need retraining with the this new argument changed:

Screenshot 2024-06-20 at 20 36 16

@Laughing-q
Copy link
Member Author

@glenn-jocher No worries! I've updated it back to reg_max.
BTW the reason why I made the update reg_max-1 to reg_max here is that I found we have everywhere reg_max equals to 16 expect the one in BboxLoss(because we are passing reg_max-1 to initialize the loss function). I figured we should keep the consistency everywhere for reg_max for better maintenance.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request TODO Items that needs completing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

8 participants