Added channel pruning on a pytorch version of YOLOv3
Use original darknet cfg and weights file as input.
It will generate pruned_cfg and pruned_weights which could be used with the original darknet.
Also compatible with AlexeyAB darknet modification.
There is already a pruned model of YOLOv3-TINY trained on VOC2007+VOC2012 in the output folder.
Check out the pruned filters in the pruned_channels.txt
forked from ayooshkathuria/pytorch-yolo-v3
-
Notifications
You must be signed in to change notification settings - Fork 0
Channel pruning on YOLOv3-TINY
Sephirot1st/pytorch-yolo-v3
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Channel pruning on YOLOv3-TINY
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%