googlenet_v3.py 文件源码

def __call__(self, x, train=False):
        pool = Inception_A.max_or_ave(self.pool)
        if self.stride == 1:
            a = self.double_convnxn_1(x, train)
            a = self.double_convnxn_2(a, train)
            a = self.double_convnxn_3(a, train)
            a = self.double_convnxn_4(a, train)
            a = self.double_convnxn_5(a, train)
            b = self.convnxn_1(x, train)
            b = self.convnxn_2(b, train)
            b = self.convnxn_3(b, train)
            c = pool(x, ksize=3, stride=self.stride, pad=1)
            c = self.conv_pool(c, train)
            d = self.conv1x1(x, train)
            return F.concat((a, b, c, d), axis=1)
        else:
            a = self.double_convnxn_1(x, train)
            a = self.double_convnxn_2(a, train)
            a = self.double_convnxn_3(a, train)
            a = self.double_convnxn_4(a, train)
            b = self.convnxn_1(x, train)
            b = self.convnxn_2(b, train)
            c = pool(x, ksize=3, stride=self.stride, pad=1)
            return F.concat((a, b, c), axis=1)