yolov5的权重文件.pt文件转化成.weights文件

在yolov5的export.py代码中添加：
from models.yolo import Model
model=Model(‘yolov5s.yaml’)
model.load_state_dict(torch.load(‘yolov5s.pt’)[‘model’].state_dict())
torch.save(model.state_dict(),‘yolov5s.weights’)
运行即可在当前目录下生成.weights文件