Unverified 提交 8c0e7092 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update README.md

上级 354109c5
......@@ -108,8 +108,7 @@ import torch
from PIL import Image
# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).fuse().eval() # yolov5s.pt
model = model.autoshape() # for autoshaping of PIL/cv2/np inputs and NMS
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).autoshape() # for PIL/cv2/np inputs and NMS
# Images
img1 = Image.open('zidane.jpg')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论