* Avoid out-of-image class labels

* Update plots.py

* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>