Skip to content

This is a fast caffe implementation of ShuffleNet.

Notifications You must be signed in to change notification settings

hshen14/ShuffleNet

 
 

Repository files navigation

ShuffleNet

This is caffe 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. This code is based on camel007's implementation(https://github.com/camel007/Caffe-ShuffleNet), but I recode the cuda file for acceleration.

How to use?

caffe.proto:

message LayerParameter {
...
optional ShuffleChannelParameter shuffle_channel_param = 164;
...
}
...
message ShuffleChannelParameter {
  optional uint32 group = 1[default = 1]; // The number of group
}

About

This is a fast caffe implementation of ShuffleNet.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 58.8%
  • Cuda 41.2%