下载 Yolov5 预训练模型

要在Python中下载Yolov5预训练模型，你可以使用`torch.hub`模块，它简化了模型下载和加载的过程。以下是步骤：

1. 首先，确保你已经安装了PyTorch和`torchvision`库。如果没有，可以使用以下命令安装：

   pip install torch torchvision

2. 导入必要的模块：

   import torch
   from torch.hub import download_url_to_file, model_zoo

3. 使用`model_zoo`下载预训练模型。以Yolov5s为例，你可以这样获取：

   url = model_zoo.url_map['yolov5s']
   filename = download_url_to_file(url, 'yolov5s.pt')

这里`'yolov5s.pt'`是模型文件名，可以根据需要替换为其他版本（如'yolov5m.pt'、'yolov5l.pt'或'yolov5x.pt'）。

4. 下载完成后，你可以加载模型：

   model = torch.load(filename, map_location='cpu')  # 注意: 如果你的设备是GPU，可能需要map_location='cuda'

注意，`model_zoo.url_map`可能会随PyTorch版本更新而改变，所以最好查看官方文档确认最新的URL。此外，如果你打算在实际项目中使用，还需要根据项目需求进行模型配置和调整。