weldon_resnet.py 文件源码

def resnet50_weldon(num_classes, pretrained=True, kmax=1, kmin=None):
    model = models.resnet50(pretrained)
    pooling = WeldonPool2d(kmax, kmin)
    return ResNetWSL(model, num_classes, pooling=pooling)