Is prediction speed of YOLOv5 recorded somewhere as part of the integration?

Hello!
I’ve been training YOLOv5 models using wandb, and I’ve been amazed at how much is natively built in with just a few command-line arguments. One thing I’m lacking - is there any recording of the speed at which it makes predictions? I don’t particularly care if it’s on the training or test sets, I’m just looking for any record of the speed at which the model makes predictions.

Thanks!
Ian

Hi,

This is probably a YoloV5 specific question, but in general, in cases like these where you want an additional logged metric, you would just record it yourself and call wandb.log({'speed': speed}, commit=false).
That being said, I had a look at the yolov5 docs and it looks like you can pass —profile to get it to profile the inference speed of your model. Hope this helps :slight_smile:

Hi Scott thanks for your reply! I’ll try using wandb.log({'speed': speed}, commit=false)

I looked in the docs here and didn’t see any reference to a -profile option, nor is it in the source code for train.py for YOLOv5. Where did you see it mentioned? Is it a different script to run other than train.py?

It seems that speed is not an option? I tried it and received an error.

----> 6 wandb.log({‘speed’: speed}, commit=false)

NameError: name ‘speed’ is not defined

I’ll reach out to the author of YoloV5 and find the appropriate command to profile the inference speed.

In the meantime, this tutorial profiles some different configurations.
https://colab.research.google.com/github/ultralytics/yolov5/blob/master/tutorial.ipynb#scrollTo=zR9ZbuQCH7FX