models.py 文件源码

def forward(self, x):
        relu1_1  = F.relu(self.conv1_1(x))
        relu1_2  = F.relu(self.conv1_2(relu1_1))
        maxpool_1  = F.max_pool2d(relu1_2, kernel_size=2, stride=2)
        relu2_1  = F.relu(self.conv2_1(maxpool_1))
        relu2_2  = F.relu(self.conv2_2(relu2_1))
        maxpool_2  = F.max_pool2d(relu2_2, kernel_size=2, stride=2)
        relu3_1  = F.relu(self.conv3_1(maxpool_2))
        relu3_2  = F.relu(self.conv3_2(relu3_1))
        relu3_3  = F.relu(self.conv3_3(relu3_2))
        maxpool_3  = F.max_pool2d(relu3_3, kernel_size=2, stride=2)
        relu4_1  = F.relu(self.conv4_1(maxpool_3))
        relu4_2  = F.relu(self.conv4_2(relu4_1))
        relu4_3  = F.relu(self.conv4_3(relu4_2))

        return [relu1_2,relu2_2,relu3_3,relu4_3]