怎么把yolo5.s.pt转换成yolov5s.torchscript.pt

你可以使用以下代码将yolo5.s.pt转换为yolov5s.torchscript.pt：

import torch
from models.experimental import attempt_load

# 加载模型
model = attempt_load('yolo5s.pt', map_location=torch.device('cpu'))

# 将模型转换为TorchScript格式
model_torchscript = torch.jit.trace(model, torch.rand(1, 3, 640, 640))

# 保存模型
torch.jit.save(model_torchscript, 'yolov5s.torchscript.pt')

其中，`yolo5.s.pt`为原始的yolov5s模型文件，`yolov5s.torchscript.pt`为转换后的TorchScript格式的模型文件。这段代码会将模型加载到CPU上，并将其转换为TorchScript格式，最后保存为文件。