一品网
  • 首页

pytorch的pixelshuffle与tensorflow的depth2space的完美等价替换


例如 C=48, shuffle 上采样4x4

将pytorch nn.PixelShuffle(4) 的输入前的卷积权重和偏置进行通道上的重排即可


        if v.shape[0] == 48:
            norder = np.arange(48).reshape([-1, 4*4]).transpose(1, 0).reshape([-1]).tolist()
            v = v[norder, ...]

深度学习

相关


双一流博士整理的计算机视觉学习路线(深度学习+传统图像处理)

深度学习模型训练阶段的加速与优化

深度学习3:波士顿房价预测(2)

深度学习--GAN学习笔记

在深度学习的视觉VISION领域数据预处理的魔法常数magic constant、黄金数值的来历及讨

崂山x牛x预备队第一次直面深度学习

关于华为服务器(样机,处理器:kunpeng920)搭建深度学习环境中的一些问题的记录

第一次深度学习

使用深度学习检测混凝土结构中的表面裂缝

go微服务框架go-micro深度学习-目录

go微服务框架go-micro深度学习-目录

第一次作业——深度学习基础

标签

一品网 冀ICP备14022925号-6