提交 c8f06948 authored 作者: Glenn Jocher's avatar Glenn Jocher

PyTorch Hub update

上级 a8147204
...@@ -5,7 +5,7 @@ Usage: ...@@ -5,7 +5,7 @@ Usage:
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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论