yolov3-channel-pruning

Update:

补充prune_utils.py. 在自己的数据集上,剪枝30%-50%, mAP不变。

requirements:

  • numpy>=1.13
  • tensorboardX
pip install tensorboardX

or

git clone https://github.com/lanpa/tensorboardX && cd tensorboardX && python setup.py install
  • albumentations
pip install albumentations

or

conda install -c conda-forge imgaug
conda install albumentations -c albumentations
  • terminaltables
pip install terminaltables
  • tqdm
  • torch
  • random
  • matplotlib
  • .......

Run

python train.py --model_def config/yolov3.cfg

python train.py --model_def config/yolov3.cfg -sr

python test_prune.py

python train.py --model_def config/prune_yolov3.cfg -pre checkpoints/prune_yolov3_ckpt.pth

Reference:

YOLOv3-model-pruning(感谢Lam1360给了很多帮助)

yolov3-network-slimming

PyTorch-YOLOv3