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

Increase plot_labels() speed #1736

Merged
merged 1 commit into from
Dec 19, 2020
Merged

Increase plot_labels() speed #1736

merged 1 commit into from
Dec 19, 2020

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Dec 19, 2020

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved label plotting in training with updated visualization libraries.

πŸ“Š Key Changes

  • Removed asynchronous thread for label plotting during training in train.py.
  • Refactored label plotting in utils/plots.py to utilize Seaborn and Pandas for generating correlograms and histograms.
  • Simplified the correlogram plot code by removing exception handling around the import and plotting code.
  • Updated scatter plots to use Seaborn's histplot for better visualization and consistency.

🎯 Purpose & Impact

  • Purpose: Enhance the data visualization during training for better insights into label distributions and relationships.
  • Impact:
    • Developers will benefit from more immediate and potentially more informative visualizations without the complexity and potential errors from asynchronous plotting.
    • Overall improvements to plotting methods can lead to easier identification of issues with data sets, potentially improving model performance.
    • Users should be aware that the update will change how plots are generated and displayed in TensorBoard and other logging tools.

@glenn-jocher
Copy link
Member Author

plot_labels() implementation suffered from speed issues due to some customizations on 2d histograms and daemon thread implementation that seemed to clash with the dataloader threading. This update should reduce label plotting time on COCO from 30 seconds to about 3 seconds.

@glenn-jocher glenn-jocher merged commit 685d601 into master Dec 19, 2020
@glenn-jocher glenn-jocher deleted the label_plots branch December 19, 2020 02:05
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
taicaile pushed a commit to taicaile/yolov5 that referenced this pull request Oct 12, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 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.

None yet

1 participant