glenn-jocher
commited on
Commit
•
c4cb7c6
1
Parent(s):
d4ea61e
Update README.md (#7686)
Browse files
README.md
CHANGED
@@ -81,15 +81,14 @@ pip install -r requirements.txt # install
|
|
81 |
<details open>
|
82 |
<summary>Inference</summary>
|
83 |
|
84 |
-
|
85 |
-
. [Models](https://github.com/ultralytics/yolov5/tree/master/models) download automatically from the latest
|
86 |
YOLOv5 [release](https://github.com/ultralytics/yolov5/releases).
|
87 |
|
88 |
```python
|
89 |
import torch
|
90 |
|
91 |
# Model
|
92 |
-
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or
|
93 |
|
94 |
# Images
|
95 |
img = 'https://ultralytics.com/images/zidane.jpg' # or file, Path, PIL, OpenCV, numpy, list
|
|
|
81 |
<details open>
|
82 |
<summary>Inference</summary>
|
83 |
|
84 |
+
YOLOv5 [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36) inference. [Models](https://github.com/ultralytics/yolov5/tree/master/models) download automatically from the latest
|
|
|
85 |
YOLOv5 [release](https://github.com/ultralytics/yolov5/releases).
|
86 |
|
87 |
```python
|
88 |
import torch
|
89 |
|
90 |
# Model
|
91 |
+
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5n - yolov5x6, custom
|
92 |
|
93 |
# Images
|
94 |
img = 'https://ultralytics.com/images/zidane.jpg' # or file, Path, PIL, OpenCV, numpy, list
|