camel007 / Caffe-ShuffleNet

This is re-implementation of "ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Caffe-ShuffleNet

This is Re-implementation of ShuffleNet, For details, please read the original paper:
"ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices" by Xiangyu Zhang et. al. 2017. If you find any bug, please drop me an email. Thanks.

Notes

group 1 group 2 group 3
1 2 3 4 5 6

Each nubmer represents a channel of the feature map

step 1: Reshape

1 2
3 4
5 6

step 2: transpose

1 3 5
2 4 6

step 3: flatten

group 1 group 2 group 3
1 3 5 2 4 6

Acknowledgement  

HolmesShuan

About

This is re-implementation of "ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices"


Languages

Language:C++ 65.7%Language:Cuda 34.3%