MCCNN.py 文件源码

def forward(self, x):
        x = self.conv1(x)
        x = F.relu(x)
        x = self.bn1(x)
        x = self.conv2(x)
        x = self.bn2(x)
        x = F.max_pool2d(x, kernel_size=(1,2), stride=(1,2), padding=(0,1))
        x = F.relu(x)
        x = self.conv3(x)
        x = self.bn3(x)
        x = F.max_pool2d(x, kernel_size=(1,2), stride=(1,2), padding=(0,1))
        x = F.relu(x)
        x = self.conv4(x)
        x = self.bn4(x)
        x = F.avg_pool2d(x, kernel_size=(1,2), stride=(1,2), padding=0)
        x = F.relu(x)
        x = x.view(-1,192)

        return x