pytorch开发一个从零开始的resnet(ResNet模型)
发布时间
阅读量:
阅读量
一、resnet的基本结构

通过细致观察可以得知,基础模块形式为


二、构建BasicBlock
本次选取LinkNet34作为网络构建的实例,首先完成基础组件BasicBlock的构建,其核心结构由两个CBR模块组成。
def BasicBlock(in_ch,out_ch,stride):
return nn.Sequential(
nn.Conv2d(in_ch, out_ch, 3, stride, padding=1, bias=False),
nn.BatchNorm2d(out_ch
全部评论 (0)
还没有任何评论哟~
