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