From 34d905dc342d171c5e8d3c382aa9cc67fdfc57ec Mon Sep 17 00:00:00 2001 From: longriyao Date: Thu, 13 Jul 2017 18:32:46 +0800 Subject: [PATCH 1/2] give the output shape of shuffle --- shuffle_channel_layer.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/shuffle_channel_layer.cpp b/shuffle_channel_layer.cpp index 9351f6c..4b3b09a 100644 --- a/shuffle_channel_layer.cpp +++ b/shuffle_channel_layer.cpp @@ -11,6 +11,7 @@ void ShuffleChannelLayer::LayerSetUp(const vector *> &bottom, group_ = this->layer_param_.shuffle_channel_param().group(); CHECK_GT(group_, 0) << "group must be greater than 0"; temp_blob_.ReshapeLike(*bottom[0]); + top[0]->ReshapeLike(*bottom[0]); } template From 76d3f04a47c987a210000f8747bd3da403023f22 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E5=9E=9A?= Date: Fri, 1 Jun 2018 20:26:45 +0800 Subject: [PATCH 2/2] Update README.md --- README.md | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index f76cef0..91b2737 100644 --- a/README.md +++ b/README.md @@ -4,9 +4,13 @@ This is Re-implementation of ShuffleNet, For details, please read the original p Neural Network for Mobile Devices" by Xiangyu Zhang et. al. 2017](https://arxiv.org/pdf/1707.01083.pdf). 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 + +| 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 @@ -15,8 +19,10 @@ Neural Network for Mobile Devices" by Xiangyu Zhang et. al. 2017](https://arxiv. 1 3 5 2 4 6　　 ## step 3: flatten - | group 1 | group 2 | group 3 | - | 1 3 | 5 2 | 4 6 | + +| group 1 | group 2 | group 3 | +| :-----: | :------: | :-------: | +| 1 3 | 5 2 | 4 6 | # Acknowledgement   [HolmesShuan](https://github.com/HolmesShuan)