修改# 加载YOLOv7模型 model = attempt_load(weights='yolov7.pt', # 模型权重文件 device='cpu') # 运行设备(CPU或GPU)报错TypeError: attempt_load() got an unexpected keyword argument 'device'
时间: 2024-03-05 17:48:53 浏览: 215
YOLOV7-模型训练代码
这个错误可能是因为你使用的是较旧版本的YOLOv7模型,该模型中没有`device`参数。您可以尝试更新YOLOv7模型或尝试使用以下代码加载模型:
```
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = torch.load('yolov7.pt', map_location=device)['model'].float()
model.to(device).eval()
```
这将从文件中加载YOLOv7模型,并将其移动到可用的GPU(如果存在)或CPU上进行推理。
阅读全文