Aller au contenu
ModelPrésentéExterne
Télécharger

Blip Cxr Mimic Finetuned

A fine-tuned version of Salesforce/blip-image-captioning-base on the MIMIC-CXR dataset for automatic chest X-ray report generation.

TypeText modelModèle de baseSalesforce/blip-image-captioning-baseVersionmodelBClasse

Source du modèle

Type
Text model
Taille du fichier
854 MB
Modèle de base
Salesforce/blip-image-captioning-base
Version
vmodel
Mis à jour
23 oct. 2026
Téléchargements
14
Source
Hugging Face

Description de la source

BLIP — Chest X-Ray Captioning (MIMIC-CXR Fine-Tune)

A fine-tuned version of Salesforce/blip-image-captioning-base on the MIMIC-CXR dataset for automatic chest X-ray report generation.

Dataset

  • Source: itsanmolgupta/mimic-cxr-dataset
  • Train split: 19,600 samples (indices 9,000 – 28,600)
  • Validation split: 2,000 samples (indices 28,600 – 30,600)

Usage

from transformers import AutoProcessor, AutoModelForImageTextToText
from PIL import Image
import torch

repo_id  = "AliFadel/blip-cxr-mimic-finetuned"
processor = AutoProcessor.from_pretrained(repo_id)
model     = AutoModelForImageTextToText.from_pretrained(repo_id)
model.eval()

image = Image.open("chest_xray.jpg").convert("RGB")
inputs = processor(images=image, text="Chest X-Ray", return_tensors="pt")

with torch.no_grad():
    ids = model.generate(
        **inputs,
        num_beams=5,
        max_length=150,
        repetition_penalty=1.5,
        top_p=0.95,
    )

caption = processor.batch_decode(ids, skip_special_tokens=True)[0]
print(caption)

Intended Use

This model is intended for research purposes only. It should not be used as a clinical diagnostic tool without expert medical supervision.

Sources

1 source
Hugging FacePrincipalVersion 44639fc49c9996de16fb2322f0fca5cd909f266b
huggingface.co/AliFadel/blip-cxr-mimic-finetuned/tree/44639fc49c9996de16fb2322f0fca5cd909f266b

Vérifié 23 oct.

Artefacts du modèle

1 artefact

model.safetensors

safetensors · 854 MB · SHA-256 bdbefb18a811…4967 · Hugging Face

Download file

Extraits de sources

3 extraits

--- language: - en license: apache-2.0 tags: - medical - radiology - chest-xray - image-captioning - blip - mimic-cxr datasets: - itsanmolgupta/mimic-cxr-dataset base_model: Salesforce/blip-image-captioning-base model_type: blip pipeline_tag: image-to-text library_name: transformers --- # BLIP — Chest X-Ray Captioning (MIMIC-CXR Fine-Tune) A fine-tuned version of [`Salesforce/blip-image-captioning-base`](https://huggingface.co/Salesforce/blip-image-captioning-base) on the **MIMIC-CXR** dataset for automatic chest X-ray report generation. ## Dataset - **Source**: [`itsanmolgupta/mimic-cxr-dataset`](https://huggingface.co/datasets/itsanmolgupta/mimic-cxr-dataset) - **Train split**: 19,600 samples (indices 9,000 – 28,600) - **Validation split**: 2,000 samples (indices 28,600 – 30,600) ## Usage ```python from transformers import AutoProcessor, AutoModelForImageTextToText from PIL import Image import torch repo_id = "AliFadel/blip-cxr-mimic-finetuned" processor = AutoProcessor.from_pretrained(repo_id) model = AutoModelForImageTextToText.from_pretrained(repo_id) model.eval() image = Image.open("chest_xray.jpg").convert("RGB") inputs = processor(images=image, text="Chest X-Ray", return_tensors="pt") with torch.no_grad(): ids = model.generate( **inputs, num_beams=5, max_length=150, repetition_penalty=1.5, top_p=0.95, ) caption = processor.batch_decode(ids, skip_special_tokens=True)[0] print(caption) ``` ## Intended Use This model is intended **for research purposes only**. It should not be used as a clinical diagnostic tool without expert medical supervision.

Source context: 14 downloads · 1 likes · Pipeline image-to-text · Library transformers · Repo AliFadel/blip-cxr-mimic-finetuned

Source context: 18 downloads · 1 likes · Pipeline image-to-text · Library transformers · Repo AliFadel/blip-cxr-mimic-finetuned