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

Simplified PyTorch hub for custom models (#1677)

上级 f7a923b4
......@@ -129,7 +129,7 @@ def custom(path_or_model='path/to/model.pt'):
if __name__ == '__main__':
model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example
# model = custom(model='path/to/model.pt') # custom example
# model = custom(path_or_model='path/to/model.pt') # custom example
model = model.autoshape() # for PIL/cv2/np inputs and NMS
# Verify inference
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论