YOLOv5 | PyTorch
pytorch.org › hub › ultralytics_yolov5Load From PyTorch Hub. This example loads a pretrained YOLOv5s model and passes an image for inference. YOLOv5 accepts URL, Filename, PIL, OpenCV, Numpy and PyTorch inputs, and returns detections in torch, pandas, and JSON output formats. See our YOLOv5 PyTorch Hub Tutorial for details.