Repository navigation
Releases
Releases · ethanhe42/channel-pruning
faster-RCNN 2X-4X Acceleration
| model | AP@.50:.05:.95 |
|---|---|
| faster-RCNN | 36.7 |
| faster-RCNN 2X | 36.7 |
| faster-RCNN 4X | 35.1 |
click the link above to download prototxts. VGGx4.v2.caffemodel is the initial model for final model vggx4_iter_70000.caffemodel.
ResNet-50 2X acceleration
we've released pruned ResNet-50 2X model:
resnet-50-cp.caffemodel, resnet-50-cp.prototxt .
Different from VGG-16 3C 4X, this model only used channel pruning algorithm.
| model | Speed-up | Top-5 | Top-1 |
|---|---|---|---|
| ResNet-50 | 2X | 90.8 | 72.3 |
| ResNet-50 | - | 92.2 | 75.3 |
Feel free to test them and leave your comments.
channel pruning 5X released
we've released pruned VGG-16 5X model:
channel_pruning.caffemodel, channel_pruning.prototxt .
Different from VGG-16 3C 4X, this model only used channel pruning algorithm.
| model | Speed-up | Top-5 | Top-1 |
|---|---|---|---|
| VGG-16 | 5x | 88.1 | 67.8 |
Feel free to test them and leave your comments.
VGG-16 3C4x caffemodel released
we've released pruned VGG-16 3C 4X model: channel_pruning_VGG-16_3C4x.caffemodel, channel_pruning_VGG-16_3C4x.prototxt .
| model | Speed-up | Top-5 | Top-1 |
|---|---|---|---|
| VGG-16 | 4x | 89.8904 | 70.574 |
Feel free to test them and leave your comments.
