7. Adjust tensor1 (8,64,32,32) to tensor2 (8,128,128,128)

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)

Guess you like

Origin blog.csdn.net/panchang199266/article/details/129850613