INNER CODE UNIT · Python

resize_image

graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:74

def resize_image(image, size):
    return image.resize(size, Image.ANTIALIAS)

def load_image(image_path, transform):
    image = Image.open(image_path)
    img = resize_image(image, [128, 32])
    img_tensor = transform(img).unsqueeze(0)
    return img_tensor


def greedy_generated_formula(image_path, model, max_len=38):
    # Prepare an image
    model.eval()
    transform = transforms.ToTensor()
    image = load_image(image_path, transform)
    image_tensor = image.to(device)
    
    with torch.no_grad():

View source record →