Advertisement

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)

还没有任何评论哟~