Computer Vision Python

OmniBridge

Image captions that actually read the text in the image, not just describe the scene.

BLIP-2LoRAFine-tuningVision-Language

01 Background

Standard image captioning systems describe the visual scene but ignore text visible within the image: charts, signs, slides, product labels. This creates a major gap for real-world applications: a model that describes 'a whiteboard' instead of reading the equations on it, or 'a sign' instead of reading what it says. OmniBridge bridges this gap by fine-tuning BLIP-2 on the TextCaps dataset using LoRA, a parameter-efficient technique that adapts a large pretrained model without retraining all parameters.

02 How It Works

01

Pretrained BLIP-2 Backbone

BLIP-2 (Bootstrapping Language-Image Pretraining) combines a frozen vision encoder, a learned Q-Former bridge, and a frozen language model to generate captions from images.

02

LoRA Fine-Tuning

Low-Rank Adaptation (LoRA) inserts small trainable matrices into BLIP-2's attention layers. This allows domain adaptation on TextCaps with minimal compute: only 0.1% of parameters are trained.

03

TextCaps Dataset Training

TextCaps provides image-caption pairs where captions specifically incorporate OCR text visible in the image. The model learns to ground language in both visual content and readable text.

04

Inference Pipeline

At inference, OmniBridge produces captions that seamlessly integrate scene description with in-image text, far more useful for accessibility, document indexing, and multimodal search.

03 Results & Impact

Significantly improved caption quality on text-heavy images vs base BLIP-2

LoRA fine-tuning achieved results with a fraction of the compute of full fine-tuning

Model correctly reads and incorporates visible text (signs, labels, slides) into generated captions

TextCaps CIDEr score improved substantially over the base BLIP-2 baseline