glenn-jocher commited on
Commit
c4cb7c6
1 Parent(s): d4ea61e

Update README.md (#7686)

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -81,15 +81,14 @@ pip install -r requirements.txt # install
81
  <details open>
82
  <summary>Inference</summary>
83
 
84
- Inference with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36)
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 yolov5m, yolov5l, yolov5x, custom
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