Improved Usage example docstrings (#9075)
* Updated Usage examples * Update detect.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update predict.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
0abae780b3
commit
06831aa9e9
10 files changed
+94
-61
No files matched your search
+2
-2
@@ -1,11 +1,11 @@
|
||||
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
||||
"""
|
||||
PyTorch Hub models https://pytorch.org/hub/ultralytics_yolov5/
|
||||
PyTorch Hub models https://pytorch.org/hub/ultralytics_yolov5
|
||||
|
||||
Usage:
|
||||
import torch
|
||||
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
||||
model = torch.hub.load('ultralytics/yolov5:master', 'custom', 'path/to/yolov5s.onnx') # file from branch
|
||||
model = torch.hub.load('ultralytics/yolov5:master', 'custom', 'path/to/yolov5s.onnx') # custom model from branch
|
||||
"""
|
||||
|
||||
import torch
|
||||
|
||||
Reference in new issue
Block a user