INNER CODE UNIT · Python
load_image
graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:77
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():
encoded_image = model.encoder(image_tensor)
# encoded_image => [1, 2, 8, 512]