save-csv command header bug

[aligh993]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Detection

Bug

The save-csv command did not write the CSV header.
For Example: $ python detect.py --weights yolov5m.pt --source 0 --save-csv
predictions.csv

Environment

No response

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[UltralyticsAssistant]

👋 Hello @aligh993, thank you for reporting this potential issue with YOLOv5 🚀!

If this is indeed a 🐛 bug, we kindly ask you to provide a minimum reproducible example (MRE). This will greatly help us investigate and resolve the issue more efficiently. An MRE typically includes a clear description of the steps taken, the exact commands or code used, example data if applicable, and the observed versus expected behavior.

Your willingness to submit a pull request is fantastic! An Ultralytics engineer will also review this issue and assist you shortly 😊.

[pderrenger]

@aligh993 thank you for reporting this issue! It seems like the CSV header might not be written when using the to_csv method. Could you verify if you are using the latest YOLOv5 version? If the issue persists, you may consider customizing the to_csv method by passing additional arguments, such as header=True. For example:

csv_result = results[0].to_csv(header=True)

If this doesn't resolve your issue, please feel free to provide a minimal reproducible example, and we'll assist further. Looking forward to your PR if you find a fix!

[aligh993]

@pderrenger thank you so much. Yes, I'm using the latest version of YOLOv5, but the issue still persists. I found a fix in 'detect.py' and requested a PR at this link: #13472

[pderrenger]

Thank you for the update, @aligh993, and for taking the initiative to submit a PR! We'll review it shortly. Your contribution is greatly appreciated and helps improve YOLOv5 for everyone. If you encounter any other issues, feel free to share them.