Yolov10 Pruning

[Pbatch]

Hello.

I have made good use of the pruning scripts for YOLOv8, and was wondering if there will be a new examples script for YOLOv10?

I am happy to collaborate (i.e. on Slack / this Issue) with anyone who wants to achieve this.

Thanks :)

[Ramzes30765]

@Pbatch Hello! Can you please share your YOLOv8 pruning full script?

[VainF]

Hi All, I'm not a YOLO expert. So, it would be great if you have some ideas about YOLO-10 pruning.

[Ramzes30765]

Hi All, I'm not a YOLO expert. So, it would be great if you have some ideas about YOLO-10 pruning.

Is it possible to update yolo8_pruning.py example?

[happyiminjay1]

There are two ways

1. Ignore the Attention Layer

      '''
      if isinstance(m, (Detect, Attention)):
                      ignored_layers.append(m)
      '''
   

2. Modify Attention Layer

        -> Remove all split Operation as it is in the yolo8_pruning.py
        -> Remove Group Convolution (which are used for depth wise Convolution)