yolov5剪枝实战5:模型剪枝和fine-tune

文章目录

    • 1. 网络剪枝
    • 2. 剪枝后的模型微调
      • 2.1 finetune训练
      • 2.2 测试
      • 2.3 性能测试

1. 网络剪枝

执行命令:

python prune.py --percent 0.5 --weights runs/train/yolov5s-sparse/weights/last.pt 
--da

你可能感兴趣的:(模型轻量化,剪枝,python,深度学习)