Function resize_image

agent_attack/attacks/utils.py:66–75

(image, size)

def resize_image(image, size):
    # Resize such that the largest dimension is `size`
    width, height = image.size
    if width > height:
        new_width = size
        new_height = int(size * height / width)
    else:
        new_height = size
        new_width = int(size * width / height)
    return image.resize((new_width, new_height))