Esempi in Python per resize

Esempio n. 1

def transform_image(image):
    image = torchvisionTransforms.resize(image, 256)
    image = torchvisionTransforms.center_crop(image, 224)
    image = torchvisionTransforms.to_tensor(image)
    image = torchvisionTransforms.normalize(image,
                                            mean=[0.485, 0.456, 0.406],
                                            std=[0.229, 0.224, 0.225])
    return image

Esempio n. 2

File: resnet_example.py Progetto: wangfeng012316/glow

def transform_image(image):
    """
    Given a PIL image, transform it to a normalized tensor for classification.
    """
    image = torchvisionTransforms.resize(image, 256)
    image = torchvisionTransforms.center_crop(image, 224)
    image = torchvisionTransforms.to_tensor(image)
    image = torchvisionTransforms.normalize(image,
                                            mean=[0.485, 0.456, 0.406],
                                            std=[0.229, 0.224, 0.225])
    return image