Unet 知识总结
1:背景
FCN 的变体,初衷解决生物医学图像,后来用于语义分割,如:卫星图像
分割,工业瑕疵检测。
2:网络结构
灰色箭头:skip-connection,用于特征融合。
红色箭头:池化-降低维度; 绿色箭头:upsample-恢复维度。
青色箭头:输出结果。
Encoder :(pytorch)
1. nn.Sequential(nn.Conv2d(in_channels,out_channels,3),
2. nn.BatchNorm2d(out_channels),
3. nn.ReLU(inplace=True))
nn.Conv2d() ,
1. nn.MaxPool2d(kernel_size=2,stride=2)
重复 5 次,最后一次无 maxpooling,将 feature map 送入 Decoder。
Decoder :(pytorch)
评论0