search

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs
https://arxiv.org/abs/2301.12900
MIT License
2.64k stars 329 forks source link

Yolov5 and Detectron R101FPN model for pruning #34

Open abhigoku10 opened 3 years ago

abhigoku10 commented 3 years ago

@JonnyKong @VainF thanks for sharing your repo , i have following query

  1. can we perform pruning on yolov5 v4 s/l/m versions of the model
  2. the detectron pretrained model like FasterRCNN-R101FPN model can it be pruned if so what are the steps to follow THanks in adavance
VainF commented 3 years ago

Hi @abhigoku10 , this tool was only tested on classification/segmentation models but I guess the example for classification models should also work on detectors. I am trying to add a new example for detection and I will let you know if there are some new results.

ithmz commented 2 years ago

Hey @VainF, any updates on adding examples for detection yet?

gkrisp98 commented 1 year ago

Hi @VainF , could you add an example for object detection ?

VainF commented 1 year ago

Hi @VainF , could you add an example for object detection ?

Please refer to the yolov7/v8 examples

https://github.com/VainF/Torch-Pruning/tree/master/benchmarks/prunability