INNER CODE UNIT · Python

greedy_generated_formula

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

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]
        
        batch_size = encoded_image.size(0)
        encoder_dim = encoded_image.size(-1)
        encoder_out = encoded_image.view(batch_size, -1, encoder_dim)
        # encoder_out => [1, 16, 512]
        
        num_pixels = encoder_out.size(1)

View source record →