七、将tensor1（8,64,32,32）调整成tensor2(8,128,128,128)

最新推荐文章于 2024-04-24 13:30:10 发布

咸鱼最牛逼

最新推荐文章于 2024-04-24 13:30:10 发布

阅读量155

点赞数

分类专栏： Pytorch 文章标签：深度学习 python 人工智能

本文链接：https://blog.csdn.net/panchang199266/article/details/129850613

版权

Pytorch 专栏收录该内容

10 篇文章 1 订阅

订阅专栏

from mmseg.ops import resize
from mmcv.cnn import ConvModule

# TODO：实现将输入shape转换成指定shape，比如（8,64,32,32）-->(8,128,256,256)
                conv = ConvModule(
                    in_channels=64,
                    out_channels=128,
                    kernel_size=1,
                    stride=1,
                    norm_cfg=self.norm_cfg,
                    act_cfg=self.act_cfg)
                out = resize(
                    input=conv(x),  # 1*1卷积，控制升降维
                    size=input_1.shape[2:],  # 尺寸，控制输出的尺寸，(256, 256)
                    mode=self.interpolate_mode,
                    align_corners=self.align_corners)