模型简介
模型特点
模型能力
使用案例
🚀 Qwen2.5-VL-3B-Instruct
Qwen2.5-VL-3B-Instruct是Qwen家族的最新成员,具备强大的视觉理解和分析能力,可处理图像、视频等多模态数据,适用于金融、商业等多个领域。
🚀 快速开始
安装依赖
Qwen2.5-VL的代码已集成在最新的Hugging face transformers中,建议使用以下命令从源代码进行安装:
pip install git+https://github.com/huggingface/transformers accelerate bitsandbytes
否则可能会遇到以下错误:
KeyError: 'qwen2_5_vl'
同时,我们提供了一个工具包,方便你更便捷地处理各种类型的视觉输入,就像使用API一样。该工具包支持base64、URL以及交错的图像和视频。可以使用以下命令进行安装:
# 强烈建议使用 `[decord]` 特性以实现更快的视频加载
pip install qwen-vl-utils[decord]==0.0.8
如果你不使用Linux系统,可能无法从PyPI安装decord
。在这种情况下,你可以使用pip install qwen-vl-utils
,它会回退到使用torchvision进行视频处理。不过,你仍然可以从源代码安装decord,以便在加载视频时使用decord。
使用🤗 Transformers进行对话
以下是一个代码片段,展示了如何使用transformers
和qwen_vl_utils
来使用对话模型:
from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, AutoProcessor
from qwen_vl_utils import process_vision_info
# 默认:将模型加载到可用设备上
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
"jarvisvasu/Qwen2.5-VL-3B-Instruct-4bit", torch_dtype="auto", device_map="auto", load_in_4bit=True
)
# 建议启用 flash_attention_2 以实现更好的加速和内存节省,特别是在多图像和视频场景中
# model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
# "jarvisvasu/Qwen2.5-VL-3B-Instruct-4bit",
# torch_dtype=torch.bfloat16,
# attn_implementation="flash_attention_2",
# device_map="auto",
# load_in_4bit=True,
# )
# 默认处理器
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-3B-Instruct")
# 模型中每张图像的视觉令牌数量默认范围是 4-16384
# 你可以根据需要设置 min_pixels 和 max_pixels,例如令牌范围为 256-1280,以平衡性能和成本
# min_pixels = 256*28*28
# max_pixels = 1280*28*28
# processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-3B-Instruct", min_pixels=min_pixels, max_pixels=max_pixels)
messages = [
{
"role": "user",
"content": [
{
"type": "image",
"image": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg",
},
{"type": "text", "text": "Describe this image."},
],
}
]
# 推理准备
text = processor.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
text=[text],
images=image_inputs,
videos=video_inputs,
padding=True,
return_tensors="pt",
)
inputs = inputs.to("cuda")
# 推理:生成输出
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)
多图像推理
# 包含多张图像和文本查询的消息
messages = [
{
"role": "user",
"content": [
{"type": "image", "image": "file:///path/to/image1.jpg"},
{"type": "image", "image": "file:///path/to/image2.jpg"},
{"type": "text", "text": "Identify the similarities between these images."},
],
}
]
# 推理准备
text = processor.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
text=[text],
images=image_inputs,
videos=video_inputs,
padding=True,
return_tensors="pt",
)
inputs = inputs.to("cuda")
# 推理
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)
视频推理
# 包含图像列表作为视频和文本查询的消息
messages = [
{
"role": "user",
"content": [
{
"type": "video",
"video": [
"file:///path/to/frame1.jpg",
"file:///path/to/frame2.jpg",
"file:///path/to/frame3.jpg",
"file:///path/to/frame4.jpg",
],
},
{"type": "text", "text": "Describe this video."},
],
}
]
# 包含本地视频路径和文本查询的消息
messages = [
{
"role": "user",
"content": [
{
"type": "video",
"video": "file:///path/to/video1.mp4",
"max_pixels": 360 * 420,
"fps": 1.0,
},
{"type": "text", "text": "Describe this video."},
],
}
]
# 包含视频URL和文本查询的消息
messages = [
{
"role": "user",
"content": [
{
"type": "video",
"video": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-VL/space_woaudio.mp4",
},
{"type": "text", "text": "Describe this video."},
],
}
]
# 在 Qwen 2.5 VL 中,帧率信息也会输入到模型中以与绝对时间对齐
# 推理准备
text = processor.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs, video_kwargs = process_vision_info(messages, return_video_kwargs=True)
inputs = processor(
text=[text],
images=image_inputs,
videos=video_inputs,
fps=fps,
padding=True,
return_tensors="pt",
**video_kwargs,
)
inputs = inputs.to("cuda")
# 推理
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)
视频URL的兼容性在很大程度上取决于第三方库的版本。详情如下表所示。如果你不想使用默认的后端,可以通过FORCE_QWENVL_VIDEO_READER=torchvision
或FORCE_QWENVL_VIDEO_READER=decord
来更改后端。
后端 | HTTP | HTTPS |
---|---|---|
torchvision >= 0.19.0 | ✅ | ✅ |
torchvision < 0.19.0 | ❌ | ❌ |
decord | ✅ | ❌ |
批量推理
# 批量推理的示例消息
messages1 = [
{
"role": "user",
"content": [
{"type": "image", "image": "file:///path/to/image1.jpg"},
{"type": "image", "image": "file:///path/to/image2.jpg"},
{"type": "text", "text": "What are the common elements in these pictures?"},
],
}
]
messages2 = [
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who are you?"},
]
# 合并消息以进行批量处理
messages = [messages1, messages2]
# 批量推理准备
texts = [
processor.apply_chat_template(msg, tokenize=False, add_generation_prompt=True)
for msg in messages
]
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
text=texts,
images=image_inputs,
videos=video_inputs,
padding=True,
return_tensors="pt",
)
inputs = inputs.to("cuda")
# 批量推理
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_texts = processor.batch_decode(
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_texts)
🤖 ModelScope
强烈建议用户(特别是中国大陆的用户)使用ModelScope。snapshot_download
可以帮助你解决下载检查点时遇到的问题。
更多使用提示
对于输入图像,支持本地文件、base64编码和URL。对于视频,目前仅支持本地文件。
# 你可以直接将本地文件路径、URL或base64编码的图像插入到文本中的所需位置
## 本地文件路径
messages = [
{
"role": "user",
"content": [
{"type": "image", "image": "file:///path/to/your/image.jpg"},
{"type": "text", "text": "Describe this image."},
],
}
]
## 图像URL
messages = [
{
"role": "user",
"content": [
{"type": "image", "image": "http://path/to/your/image.jpg"},
{"type": "text", "text": "Describe this image."},
],
}
]
## Base64编码的图像
messages = [
{
"role": "user",
"content": [
{"type": "image", "image": "data:image;base64,/9j/..."},
{"type": "text", "text": "Describe this image."},
],
}
]
调整图像分辨率以提升性能
模型支持广泛的分辨率输入。默认情况下,它使用原始分辨率进行输入,但更高的分辨率可以提高性能,但会增加计算量。用户可以设置最小和最大像素数,以实现适合自己需求的最佳配置,例如令牌数量范围为256-1280,以平衡速度和内存使用。
min_pixels = 256 * 28 * 28
max_pixels = 1280 * 28 * 28
processor = AutoProcessor.from_pretrained(
"Qwen/Qwen2.5-VL-3B-Instruct", min_pixels=min_pixels, max_pixels=max_pixels
)
此外,我们提供了两种方法来对输入到模型的图像大小进行细粒度控制:
- 定义min_pixels和max_pixels:图像将被调整大小,以在min_pixels和max_pixels范围内保持其纵横比。
- 指定确切的尺寸:直接设置
resized_height
和resized_width
。这些值将被四舍五入到最接近的28的倍数。
# min_pixels 和 max_pixels
messages = [
{
"role": "user",
"content": [
{
"type": "image",
"image": "file:///path/to/your/image.jpg",
"resized_height": 280,
"resized_width": 420,
},
{"type": "text", "text": "Describe this image."},
],
}
]
# resized_height 和 resized_width
messages = [
{
"role": "user",
"content": [
{
"type": "image",
"image": "file:///path/to/your/image.jpg",
"min_pixels": 50176,
"max_pixels": 50176,
},
{"type": "text", "text": "Describe this image."},
],
}
]
处理长文本
当前的config.json
设置的上下文长度最大为32,768个令牌。为了处理超过32,768个令牌的大量输入,我们使用了YaRN技术,这是一种增强模型长度外推能力的技术,确保在长文本上的最佳性能。
对于受支持的框架,你可以在config.json
中添加以下内容以启用YaRN:
{
...,
"type": "yarn",
"mrope_section": [
16,
24,
24
],
"factor": 4,
"original_max_position_embeddings": 32768
}
需要注意的是,这种方法对时间和空间定位任务的性能有显著影响,因此不建议使用。同时,对于长视频输入,由于MRoPE本身在ids方面更节省,因此可以直接将max_position_embeddings修改为更大的值,例如64k。
✨ 主要特性
关键增强功能
- 视觉理解能力:Qwen2.5-VL不仅擅长识别常见物体,如花鸟鱼虫,还能够高度分析图像中的文本、图表、图标、图形和布局。
- 智能代理能力:Qwen2.5-VL可直接作为视觉代理,进行推理并动态指导工具,具备计算机和手机使用能力。
- 长视频理解与事件捕捉:Qwen2.5-VL能够理解超过1小时的视频,并具备通过精确确定相关视频片段来捕捉事件的新能力。
- 多格式视觉定位:Qwen2.5-VL可以通过生成边界框或点来准确地在图像中定位对象,并能为坐标和属性提供稳定的JSON输出。
- 结构化输出生成:对于发票、表格、表单等扫描数据,Qwen2.5-VL支持对其内容进行结构化输出,有利于金融、商业等领域的应用。
模型架构更新
- 用于视频理解的动态分辨率和帧率训练:通过采用动态FPS采样,将动态分辨率扩展到时间维度,使模型能够理解不同采样率的视频。相应地,在时间维度上用ids和绝对时间对齐更新mRoPE,使模型能够学习时间序列和速度,最终获得精确确定特定时刻的能力。
- 精简高效的视觉编码器:通过在ViT中策略性地实现窗口注意力,提高了训练和推理速度。ViT架构进一步通过SwiGLU和RMSNorm进行了优化,使其与Qwen2.5 LLM的结构保持一致。
我们有三个分别具有30亿、70亿和720亿参数的模型。本仓库包含经过指令微调的30亿参数的Qwen2.5-VL模型。更多信息,请访问我们的博客和GitHub。
📚 详细文档
图像基准测试
基准测试 | InternVL2.5-4B | Qwen2-VL-7B | Qwen2.5-VL-3B |
---|---|---|---|
MMMUval | 52.3 | 54.1 | 53.1 |
MMMU-Proval | 32.7 | 21.2 | 31.6 |
AI2Dtest | 81.4 | 83.0 | 81.5 |
DocVQAtest | 91.6 | 94.5 | 93.9 |
InfoVQAtest | 72.1 | 76.5 | 77.1 |
TextVQAval | 76.8 | 84.3 | 79.3 |
MMBench-V1.1test | 79.3 | 80.7 | 77.6 |
MMStar | 58.3 | 60.7 | 55.9 |
MathVistatestmini | 60.5 | 58.2 | 62.3 |
MathVisionfull | 20.9 | 16.3 | 21.2 |
视频基准测试
基准测试 | InternVL2.5-4B | Qwen2-VL-7B | Qwen2.5-VL-3B |
---|---|---|---|
MVBench | 71.6 | 67.0 | 67.0 |
VideoMME | 63.6/62.3 | 69.0/63.3 | 67.6/61.5 |
MLVU | 48.3 | - | 68.2 |
LVBench | - | - | 43.3 |
MMBench-Video | 1.73 | 1.44 | 1.63 |
EgoSchema | - | - | 64.8 |
PerceptionTest | - | - | 66.9 |
TempCompass | - | - | 64.4 |
LongVideoBench | 55.2 | 55.6 | 54.2 |
CharadesSTA/mIoU | - | - | 38.8 |
代理基准测试
基准测试 | Qwen2.5-VL-3B |
---|---|
ScreenSpot | 55.5 |
ScreenSpot Pro | 23.9 |
AITZ_EM | 76.9 |
Android Control High_EM | 63.7 |
Android Control Low_EM | 22.2 |
AndroidWorld_SR | 90.8 |
MobileMiniWob++_SR | 67.9 |
🔧 技术细节
动态分辨率和帧率训练
通过采用动态FPS采样,将动态分辨率扩展到时间维度,使模型能够理解不同采样率的视频。相应地,在时间维度上用ids和绝对时间对齐更新mRoPE,使模型能够学习时间序列和速度,最终获得精确确定特定时刻的能力。
精简高效的视觉编码器
通过在ViT中策略性地实现窗口注意力,提高了训练和推理速度。ViT架构进一步通过SwiGLU和RMSNorm进行了优化,使其与Qwen2.5 LLM的结构保持一致。
📄 许可证
本项目采用qwen-research许可证。
📖 引用
如果您觉得我们的工作有帮助,请引用我们的工作:
@misc{qwen2.5-VL,
title = {Qwen2.5-VL},
url = {https://qwenlm.github.io/blog/qwen2.5-vl/},
author = {Qwen Team},
month = {January},
year = {2025}
}
@article{Qwen2VL,
title={Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolution},
author={Wang, Peng and Bai, Shuai and Tan, Sinan and Wang, Shijie and Fan, Zhihao and Bai, Jinze and Chen, Keqin and Liu, Xuejing and Wang, Jialin and Ge, Wenbin and Fan, Yang and Dang, Kai and Du, Mengfei and Ren, Xuancheng and Men, Rui and Liu, Dayiheng and Zhou, Chang and Zhou, Jingren and Lin, Junyang},
journal={arXiv preprint arXiv:2409.12191},
year={2024}
}
@article{Qwen-VL,
title={Qwen-VL: A Versatile Vision-Language Model for Understanding, Localization, Text Reading, and Beyond},
author={Bai, Jinze and Bai, Shuai and Yang, Shusheng and Wang, Shijie and Tan, Sinan and Wang, Peng and Lin, Junyang and Zhou, Chang and Zhou, Jingren},
journal={arXiv preprint arXiv:2308.12966},
year={2023}
}









