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)
七、将tensor1(8,64,32,32)调整成tensor2(8,128,128,128)
最新推荐文章于 2024-09-27 22:55:17 发布