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给了很多帮助)