glenn-jocher
commited on
Commit
•
d17b45e
1
Parent(s):
efe60b5
Update README.md (#4143)
Browse files
README.md
CHANGED
@@ -82,10 +82,10 @@ Inference with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/is
|
|
82 |
import torch
|
83 |
|
84 |
# Model
|
85 |
-
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5m, yolov5x, custom
|
86 |
|
87 |
# Images
|
88 |
-
img = 'https://ultralytics.com/images/zidane.jpg' # or
|
89 |
|
90 |
# Inference
|
91 |
results = model(img)
|
|
|
82 |
import torch
|
83 |
|
84 |
# Model
|
85 |
+
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5m, yolov5l, yolov5x, custom
|
86 |
|
87 |
# Images
|
88 |
+
img = 'https://ultralytics.com/images/zidane.jpg' # or PosixPath, PIL, OpenCV, numpy, list
|
89 |
|
90 |
# Inference
|
91 |
results = model(img)
|