PyTorch 实现经典模型8：FCN

FCN

网络结构

代码

class fcn(nn.Module):def __init__(self, num_classes):super(fcn, self).__init__()self.stage1 = nn.Sequential(*list(pretrained_net.children())[:-4]) # 第一段self.stage2 = list(pretrained_net.children())[-4] # 第二段self.stage3 = list(pretrained_net.children())[-3] # 第三段self.scores1 = nn.Conv2d(512, num_classes, 1)self.scores2 = nn.Conv2d(256, num_classes, 1)self.scores3 = nn.Conv2d(128, num_classes, 1)self.upsample_8x = nn.ConvTranspose2d(num_classes, num_classes, 16, 8, 4, bias=False)self.upsample_8x.weight.data = bilinear_kernel(num_classes, num_classes, 16) # 使用双线性 kernelself.upsample_4x = nn.ConvTranspose2d(num_classes, num_classes, 4, 2, 1, bias=False)self.upsample_4x.weight.data = bilinear_kernel(num_classes, num_classes, 4) # 使用双线性 kernelself.upsample_2x = nn.ConvTranspose2d(num_classes, num_classes, 4, 2, 1, bias=False)   self.upsample_2x.weight.data = bilinear_kernel(num_classes, num_classes, 4) # 使用双线性 kerneldef forward(self, x):x = self.stage1(x)s1 = x # 1/8x = self.stage2(x)s2 = x # 1/16x = self.stage3(x)s3 = x # 1/32s3 = self.scores1(s3)s3 = self.upsample_2x(s3)s2 = self.scores2(s2)s2 = s2 + s3s1 = self.scores3(s1)s2 = self.upsample_4x(s2)s = s1 + s2s = self.upsample_8x(s2)return s

Ref

https://github.com/Jichao-Zhao/Semantic_Segmentation_Exer
FCN 的简单实现

PyTorch 实现经典模型8：FCN相关推荐

PyTorch 实现经典模型2：AlexNet
AlexNet 网络结构论文总结成功使用ReLU作为CNN的激活函数,并验证效果在较深的网络中超过了Sigmoid.成功解决了Sigmoid在网络较深时的梯度弥散问题. 训练时使用Dropout随 ...
PyTorch 实现经典模型3：VGG
VGG 网络结构代码 1) 导入必需的包 # 1) 导入必需的包 import torch import torch.nn as nn import torch.nn.functional as F ...
PyTorch 实现经典模型6：RCNN (Fast RCNN, Faster RCNN)
RCNN (Fast RCNN, Faster RCNN) 适用范围图像分类(Image classification) 目标检测(Object detection) 网络结构代码 Ref R-C ...
PyTorch 实现经典模型5：ResNet
ResNet 网络结构代码 #------------------------------用50行代码搭建ResNet---------------------------------------- ...
PyTorch 实现经典模型4：GoogLeNet
GoogLeNet 创新点: 通过多种卷积核叠加网络复杂性学习多种卷积提升计算效率 GoogLeNet网络深度达到22层 1x1卷积降低维度降低计算量,提升计算效率网络结构代码 import ...
PyTorch 实现经典模型1：LeNet5
模型:LeNet5 网络结构符号说明网络参数代码实现 1) 导入必需的包 2) 搭建网络模型 3) 导入使用的数据集 4) 训练模型 5) 保存模型 6) 测试模型效果所遇错误 '_Incom ...
PyTorch 实现经典模型7：YOLO (v1, v2, v3, v4)
YOLO (v1, v2, v3, v4) 网络结构 YOLO v3 网络结构代码 Ref <机器爱学习>YOLO v1深入理解 <机器爱学习>YOLOv2 / YOLO90 ...
PyTorch Hub发布获Yann LeCun强推！一行代码调用经典模型
作者 | Team PyTorch 译者 | Monanfei 责编 | 夕颜出品 | AI科技大本营(ID: rgznai100) 导读:6月11日,Facebook PyTorch 团队推出了全 ...
Facebook 发布 PyTorch Hub：一行代码实现经典模型调用！
作者 | Team PyTorch 译者 | Monanfei 责编 | 夕颜出品 | AI科技大本营(ID: rgznai100) 6月11日,Facebook PyTorch 团队推出了全新 A ...

PyTorch 实现经典模型8：FCN

FCN

网络结构

代码

Ref

PyTorch 实现经典模型8：FCN相关推荐

最新文章

热门文章