地平线征程5-Yolo模型部署-pt转bin

一、ONNX 模型转换

1、下载yolov7

GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

下载yolov7.pt权重文件和yolov7代码。

2、打开model/yolo.pt文件

在文件中搜索 permute，一共有8处，这里是修改模型输出节点维度信息。

其中7处按照以下修改

#x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous()
x[i] = x[i].permute(0, 2, 3, 1).contiguous()

1处按照以下修改：

# x[i+self.nl] = x[i+self.nl].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous()
x[i+self.nl] = x[i+self.nl].permute(0, 2, 3, 1).contiguous()

3、打开 export.py文件,将opset_version=12中的12替换为11