YOLOV8转出的onnx模型支持腾讯NCNN架构需要修改的地方_android ncnn对于yolov8-pose模型怎么改py文件

1.修改ultralytics\nn\modules\block.py

找到类：class C2f(nn.Module):

修改函数：def forward(self, x):

将原来的

    def forward(self, x):
        """Forward pass through C2f layer."""
        y = list(self.cv1(x).chunk(2, 1))
        y.extend(m(y[-1]) for m in self.m)
        return self.cv2(torch.cat(y, 1))               

 修改为：

    def forward(self, x):
        """Forward pass through C2f layer."""
        # y = list(self.cv1(x).chunk(2, 1))
        # y.extend(m(y[-1]) for m in self.m)
        # return self.cv2(torch.cat(y, 1))
        x = self.cv1(x)
        x = [x, x[:, self.c:, ...]]
        x.extend(m(x[-1]) for m in self.m)
        x.pop(1)
        return self.cv2(torch.cat(x,1))