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

add smoothing line to results.png to improve readability #11535

Conversation

PetervanLunteren
Copy link
Contributor

@PetervanLunteren PetervanLunteren commented May 15, 2023

This PR contains a very small addition to the original code (+ 2 lines) and does not interfere with the machine learning process whatsoever. It adds a smoothing line on top of the original plots in the results.png. With this smoothing line I hope to improve plot readability for users and provide a better insight into when their model has the best fit, and where their model is under- or overfitted.

Before:
before

After:
after

🤖 Generated by Copilot at e32f5ee

Summary

📈🔬🎨

This pull request enhances the plot_results function in utils/plots.py by applying a Gaussian filter to smooth the plot lines. This makes the results easier to visualize and compare.

To plot results with more finesse
We use gaussian_filter1d to smooth and compress
The lines become clearer
And the peaks are nearer
This feature is a great success

Walkthrough

  • Import and apply gaussian_filter1d function to smooth plot lines in plot_results (link, link)

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improvement in visualization of training results with smoothed lines.

📊 Key Changes

  • Imported gaussian_filter1d from scipy.ndimage.filters.
  • Added smoothed line plots in the plot_results function using Gaussian filter.

🎯 Purpose & Impact

  • Purpose: To provide a clearer visual representation of the model's training performance over time by smoothing out the results.
  • Impact: Users can more easily identify trends and patterns in the results, as the smoothing helps to reduce noise in the visualization. This can be particularly helpful when presenting results to non-expert stakeholders or for better assessing the model's behavior during training.

PetervanLunteren and others added 2 commits May 15, 2023 09:13
Signed-off-by: Peter van Lunteren <85185478+PetervanLunteren@users.noreply.github.com>
@PetervanLunteren PetervanLunteren deleted the add_smoothing_line_to_results_png branch May 15, 2023 16:46
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