search

HankYe / PAGCP

PAGCP for the compression of YOLOv5
GNU General Public License v3.0
111 stars 11 forks source link

yolov8 pruning #15

Open Zzzxi opened 10 months ago

Zzzxi commented 10 months ago

Hi, will you consider coming out with yolov8 pruning as a follow-up?

HankYe commented 9 months ago

Hi, will you consider coming out with yolov8 pruning as a follow-up?

Hi, thanks for your attention to our work. To my knowledge, yolov8 is a bit different from yolov5 for the prunable structures but may still be within the capability of our method. So, if you are interested in pruning yolov8, you could have a try. For us, honestly, we are a bit busy with the coming work these days, so we may validate the effectiveness of pruning yolov8 at the end of this year. Once the experiment is conducted, we will update the comment. At last, thanks again for your interest in our work. Let me know if you have any other questions.

Zzzxi commented 9 months ago

I will try yolov8's pruning, looking forward to your code updates!

Hrayo712 commented 7 months ago

any update on yolov8 pruning :) ?

xiaoshuomin commented 3 months ago

@HankYe 作者有yolov8剪枝相关的更新吗？目前使用该代码对yolov8剪枝出现报错，而对yolov5剪枝正常运行

HankYe commented 2 months ago

Hi, we have tested the effectiveness of PAGCP on YOLOv8, and found that the bug comes from the wrong detection of the gradient-based dependency graph for C2f modules. One alternative way to pruning this module is customizing the pruning handler for this module in dependency.py. We will implement this method in the next version after finishing the current work.