最新消息: USBMI致力于为网友们分享Windows、安卓、IOS等主流手机系统相关的资讯以及评测、同时提供相关教程、应用、软件下载等服务。

解决U

IT圈 admin 1浏览 0评论

解决U

在U-net下采样后时,通过polling层,可能会出现这种情况,37*37feature maps 压缩成18*18大小,但在上采样过程中,利用

nn.ConvTranspose2d()通常变为36*36大小的feature maps,不同大小的feature maps在进行concat时会报错。

解决这种问题可以采用追加一个反卷积过程,在判断到两个过程得到的feature maps大小不匹配时,使用kernel_size=2,stride=1的反卷积过程,将当前特征映射扩大。

nn.ConvTranspose2d(in_channels=in_channels,out_channels=out_channels,kernel_size=2,stride=1)

解决U

在U-net下采样后时,通过polling层,可能会出现这种情况,37*37feature maps 压缩成18*18大小,但在上采样过程中,利用

nn.ConvTranspose2d()通常变为36*36大小的feature maps,不同大小的feature maps在进行concat时会报错。

解决这种问题可以采用追加一个反卷积过程,在判断到两个过程得到的feature maps大小不匹配时,使用kernel_size=2,stride=1的反卷积过程,将当前特征映射扩大。

nn.ConvTranspose2d(in_channels=in_channels,out_channels=out_channels,kernel_size=2,stride=1)

与本文相关的文章

发布评论

评论列表 (0)

  1. 暂无评论