Logger step fix: Increment step with epochs

[AyushExel]

Wandb logger will now respect the epochs count. But the logs will be delayed by one epoch, i.e, you'll only see the logs of epoch 1 when logger is called with epoch 2 and so on.

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Minor code refactoring and improvement in metrics logging for better tracking and visualization.

📊 Key Changes

• Reformatted import statements for better code readability.
• Updated weighted bi-directional map in image transformation, ensuring consistency in style.
• Ensured that Weights & Biases (wandb) logging includes the epoch as the step parameter for more precise metric tracking.

🎯 Purpose & Impact

• Code Readability: Adjusted code style leads to better maintainability and readability, making it easier for developers to navigate and contribute.
• Image Transformation Clarity: The adjustment in the image transpose dictionary improves the code presentation without changing functionality.
• Enhanced Tracking: Including epoch in wandb logs allows users to better track model performance over time, which is crucial for experiment tracking and comparison across training runs. This results in a more intuitive visualization of training progress in wandb dashboards.

[AyushExel]

Pre commit applied many changes. I guess its configured to not check your commits

[glenn-jocher]

@AyushExel PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐