通道随机混合操作（Channel Shuffle Operation）

最新推荐文章于 2025-03-27 20:32:27 发布

鹊踏枝-码农

最新推荐文章于 2025-03-27 20:32:27 发布

阅读量9.5k

点赞数 12

分类专栏： PyTorch 文章标签： ShuffleNet Channel Shuffle 通道随机混合

本文链接：https://blog.youkuaiyun.com/u011501388/article/details/96322572

版权

PyTorch 专栏收录该内容

14 篇文章

订阅专栏

通道随机混合操作（Channel Shuffle Operation）可以看成“重塑-转置-重塑”（“reshapetranspose-
reshape”）操作。这里假设把4个Feature Maps级联后，共1024个Channels。现在我们想把这个1024个Channels随机打乱混合。首先把Channels重塑为（g, c），其中 g 表示分组数目，c=1024/g。然后把它转置一下为（c, g）。然后把它重塑为1024个通道。

具体代码如下：

import torch


def channel_shuffle(x, groups):
    batchsize, num_channels, height, width = x.data.size()

    channels_per_group = num_channels // groups

    # reshape
    x = x.view(batchsize, groups,
               channels_per_group, height, width)

    # transpose
    # - contiguous() required if transpose() is used before view().
    #   See https://github.com/pytorch/pytorch/issues/764
    x = torch.transpose(x, 1, 2).contiguous()

    # flatten
    x = x.view(batchsize, -1, height, width)


    return x

如果您觉得我的文章对您有所帮助，欢迎扫码进行赞赏！