yolov3-network-slimming
将Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017)应用在yolov3上
环境
pytorch 0.4
window 10
如何使用
1.对原始weights文件进行稀疏化训练
python sparsity_train.py -sr --s 0.0001 --image_folder coco.data --cfg yolov3.cfg --weights yolov3.weights
2.剪枝
python prune.py --cfg yolov3.cfg --weights checkpoints/yolov3_sparsity_100.weights --percent 0.3
3.对剪枝后的weights进行微调
python sparsity_train.py --image_folder coco.data --cfg prune_yolov3.cfg --weights prune_yolov3.weights