def vgg_weldon(num_classes, pretrained=True, kmax=1, kmin=None):
model = models.vgg19(pretrained)
pooling = WeldonPool2d(kmax, kmin)
return VGGWSL(model, num_classes, pooling=pooling)
def resnet152_weldon(num_classes, pretrained=True, kmax=1, kmin=None):
model = models.resnet152(pretrained)
pooling = WeldonPool2d(kmax, kmin)
return ResNetWSL(model, num_classes, pooling=pooling)
def densenet121_weldon(num_classes, pretrained=True, kmax=1, kmin=None):
model = models.densenet121(pretrained)
pooling = WeldonPool2d(kmax, kmin)
return DenseNetWSL(model, num_classes, pooling=pooling)