Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
hubconf.py
View File
| model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) | model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) | ||||
| """ | """ | ||||
| dependencies = ['torch', 'pyyaml'] | |||||
| dependencies = ['torch', 'yaml'] | |||||
| import torch | import torch | ||||
| from models.yolo import Model | from models.yolo import Model |
Loading…