Glenn Jocher
GitHub
3 years ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
README.md
View File
| import torch | import torch | ||||
| # Model | # Model | ||||
| model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5m, yolov5x, custom | |||||
| model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # or yolov5m, yolov5l, yolov5x, custom | |||||
| # Images | # Images | ||||
| img = 'https://ultralytics.com/images/zidane.jpg' # or file, PIL, OpenCV, numpy, multiple | |||||
| img = 'https://ultralytics.com/images/zidane.jpg' # or PosixPath, PIL, OpenCV, numpy, list | |||||
| # Inference | # Inference | ||||
| results = model(img) | results = model(img) |
Loading…