如何將YOLOv5輸出三個output

想請教版上各位
我是要使用ultralytics/yolov5的yolov5s的模型去修改
請問一下，我該如何使用已訓練好的模型，僅保留到yolo層，移除掉最後的concat呢？

謝謝！

• 这篇博客: YOLOv5模型剪枝压缩(2)-YOLOv5模型简介和剪枝层选择中的 6. Concat 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

• 拼接层，用于拼接之前的层，例如

     [[-1, 6], 1, Concat, [1]],  # cat backbone P4

  其中[-1, 6]是将上一层和第6个模块输出进行通道拼接。