资源算法ShuffleNet_V2-caffe

ShuffleNet_V2-caffe

2020-01-16 | |  53 |   0 |   0

This is caffe implementation of shuffleNet V2, For details, you can read the original paper:

"ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design"

This model's shuffle_channel cuda file for acceleration is based on farmingyard/ShuffleNet.  conv_dw_layer cuda file is based on farmingyard/caffe-mobilenet

Note: If you want to train Shufflenet V2 on caffe, firstly, you should add all of *.cu and *.cpp to src/caffe/layers, and add all of *.hpp to include/caffe/layers . Then, you need change the caffe.proto :

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

If you have no clue about how to add the ShuffleChannel layer and ConvolutionDepthwise layer, take this blog"ShuffleNet在Caffe框架下的实现" as a reference.


上一篇:ShuffleNetV2-pytorch

下一篇:shufflenet-ssd

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • Klukshu-Sockeye-...

    KLUKSHU SOCKEYE PROJECTS 2016 This repositor...