model = models.yolov5s(pretrained=True) AttributeError: module 'torchvision.models' has no attribute 'yolov5s'
时间: 2023-11-07 14:48:35 浏览: 192
yolov5s模型没有前面的slice模块
这个错误可能是因为您使用的 `torchvision` 版本过低,不支持 `yolov5s` 模型。您可以尝试更新 `torchvision` 到最新版本,或者手动下载 `yolov5s` 模型并加载它。
如果您想手动下载 `yolov5s` 模型,可以使用以下代码:
```python
import torch
model_url = "https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt"
model_path = "yolov5s.pt"
torch.hub.download_url_to_file(model_url, model_path)
model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path)
```
这将从 Github 上下载 `yolov5s` 模型,并将其保存到当前目录下的 `yolov5s.pt` 文件中。然后,您可以使用 `torch.hub.load()` 加载模型。
阅读全文