https://github.com/farmingyard/ShuffleNet

This is a fast caffe implementation of ShuffleNet.
https://github.com/farmingyard/ShuffleNet

caffe shufflenet shufflenet-v2

Last synced: about 1 month ago
JSON representation

This is a fast caffe implementation of ShuffleNet.

• Host: GitHub
• URL: https://github.com/farmingyard/ShuffleNet
• Owner: farmingyard
• Created: 2017-07-15T13:22:03.000Z (over 7 years ago)
• Default Branch: master
• Last Pushed: 2018-08-30T04:17:48.000Z (over 6 years ago)
• Last Synced: 2024-08-01T22:50:08.607Z (4 months ago)
• Topics: caffe, shufflenet, shufflenet-v2
• Language: C++
• Homepage:
• Size: 15.6 KB
• Stars: 449
• Watchers: 32
• Forks: 246
• Open Issues: 17
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-image-classification - unofficial-caffe : https://github.com/farmingyard/ShuffleNet
• awesome-image-classification - unofficial-caffe : https://github.com/farmingyard/ShuffleNet

README

        
# 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](https://arxiv.org/pdf/1707.01083.pdf).

This code is based on camel007's implementation(https://github.com/camel007/Caffe-ShuffleNet), but I recode the cuda file for acceleration.

#### Update: add caffe deploy file of shufflenet v2 1x.

## 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

}

```