Yolov5 export torchscript

要将 YOLOv5 导出为 TorchScript 模型，您可以按照以下步骤操作：

1. 安装 YOLOv5

首先，您需要安装 YOLOv5。您可以从 Github 上的官方 YOLOv5 仓库中获取代码，并按照说明进行安装。

2. 下载预训练模型

您需要下载预训练的 YOLOv5 模型。您可以从官方仓库中下载训练好的模型，并将其放在某个目录下。

3. 加载模型

使用 PyTorch，您可以使用以下代码加载模型：

import torch

model = torch.hub.load('ultralytics/yolov5', 'custom', path='path/to/weights.pt')

这将加载训练好的 YOLOv5 模型，并将其转换为 TorchScript 模型。

4. 导出 TorchScript 模型

使用以下代码，您可以将模型导出为 TorchScript 格式：

model.model[-1].export = True  # set Detect() layer export=True
x = torch.rand(1, 3, 640, 640)
model(x)  # dry run
model.model[-1].export = False  # reset
model = model.model[-1].torchscript()

这将导出最后一层检测层，并将模型转换为 TorchScript 格式。

5. 保存模型

最后，您可以使用以下代码保存 TorchScript 模型：

torch.jit.save(model, 'yolov5s.torchscript.pt')

这将保存 TorchScript 模型到本地文件中。

这就是将 YOLOv5 导出为 TorchScript 模型的步骤。