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

Model enhancement #75

Merged
merged 5 commits into from
Dec 14, 2022
Merged

Model enhancement #75

merged 5 commits into from
Dec 14, 2022

Conversation

AyushExel
Copy link
Contributor

@AyushExel AyushExel commented Dec 14, 2022

Adds:

  • model.fuse()
  • model.val()
  • model.visualize_preds()
  • model.predict()
  • improves resume
    TODO:
  • Add pytest-like model tests
  • Write docs

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements to the YOLO model testing and loading capabilities.

πŸ“Š Key Changes

  • Added new test cases for various model functionalities.
  • Expanded the YOLO class with methods info, fuse, predict, val, resume.
  • Introduced model-task associations enabling clearer operations based on model types.
  • Enhanced flexibility in handling model files and inputs.
  • Added smart inference using the @smart_inference_mode decorator.

🎯 Purpose & Impact

  • 🎨 Streamlines the testing process by modularizing tests and adding new ones for major functionalities.
  • πŸš€ Enables a more robust usage of the YOLO model class with improved methods for info logging, model fusion, prediction visualization, validation, and resumption of training.
  • 🌐 Fosters ease of integration with diverse model formats and devices for wider applicability and superior performance.
  • πŸ§‘β€πŸ’» Provides developers with more intuitive control over tasks and models while enhancing the usability of the model for end users across various applications.

@AyushExel
Copy link
Contributor Author

AyushExel commented Dec 14, 2022

@Laughing-q I've added a model tester file that tests all functionality of the model with one script. Once we have the assets in place we can then start using it in the ci.

I'll merge it like this wihtout adding preds post-process as I also cannot see how users will use it. I think .predict is what we want users to use

@AyushExel AyushExel merged commit eb5adf4 into main Dec 14, 2022
@AyushExel AyushExel deleted the model_enhance branch December 14, 2022 09:03
0iui0 pushed a commit to 0iui0/ultralytics that referenced this pull request Jan 3, 2024
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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.

None yet

1 participant