Skip to content

Commit

Permalink
Update ultralytics_yolov5.md
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jan 27, 2021
1 parent 8bcd4b6 commit 78a5f7a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ultralytics_yolov5.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ YOLOv5 is a family of compound-scaled object detection models trained on the COC

## Load From PyTorch Hub

This simple example loads a pretrained YOLOv5s model from PyTorch Hub as `model` and passes two image URLs for batched inference.
This simple example loads a pretrained **YOLOv5s** model from PyTorch Hub as `model` and passes two **image URLs** for batched inference.

```python
import torch
Expand All @@ -71,7 +71,7 @@ print(results.xyxy[0]) # print img1 predictions (pixels)
# [4.24503e+02, 4.29092e+02, 5.16300e+02, 7.16425e+02, 5.68713e-01, 2.70000e+01]])
```

To load YOLOv5 from PyTorch Hub for inference with PIL, OpenCV, Numpy or PyTorch inputs please see the full [YOLOv5 PyTorch Hub Tutorial](https://github.com/ultralytics/yolov5/issues/36).
For YOLOv5 PyTorch Hub inference with **PIL**, **OpenCV**, **Numpy** or **PyTorch** inputs please see the full [YOLOv5 PyTorch Hub Tutorial](https://github.com/ultralytics/yolov5/issues/36).


## Citation
Expand Down

0 comments on commit 78a5f7a

Please sign in to comment.