海思NNIE开发·yolov8

官方预训练模型转换

1. 下载yolov8源码解压到本地，并配置基础运行环境。
2. 下载官方预训练模型

• yolov8n.pt
• yolov8s.pt
• yolov8m.pt
• …

3. 进入ultralytics-main目录下，新建文件夹weights，并将步骤2中下载的权重文件放进去。
4. 修改ultralytics/nn/modules/head.py文件

    def forward(self, x):
        """Perform forward pass through YOLO model and return predictions."""
        if torch.onnx.is_in_onnx_export():
            results = []
            for i in range(self.nl):
                results.append(nn.Flatten()(self.cv2[i](x[i]).permute(0, 2, 3, 1).contiguous()))
                results.append(nn.Flatten