Skip to content

Latest commit

 

History

History
215 lines (162 loc) · 28.3 KB

README_zh-CN.md

File metadata and controls

215 lines (162 loc) · 28.3 KB

VLMEvalKit (python 包名为 vlmeval) 是一款专为大型视觉语言模型 (Large Vision-Language Models, LVLMs) 评测而设计的开源工具包。该工具支持在各种基准测试上对大型视觉语言模型进行一键评估,无需进行繁重的数据准备工作,让评估过程更加简便。在 VLMEvalKit 中,我们对所有大型视觉语言模型生成的结果进行评测,并提供基于精确匹配与基于 LLM 的答案提取两种评测结果。

🆕 更新

  • [2024-11-21] 集成了一个新的配置系统,以实现更灵活的评估设置。查看文档或运行python run.py --help了解更多详情 🔥🔥🔥
  • [2024-11-21] 支持 QSpatial,一个用于定量空间推理的多模态基准(例如,确定大小/距离),感谢 andrewliao11 提供官方支持 🔥🔥🔥
  • [2024-11-21] 支持 MM-Math,一个包含约6K初中多模态推理数学问题的新多模态数学基准。GPT-4o-20240806在该基准上达到了22.5%的准确率 🔥🔥🔥
  • [2024-11-16] 支持 OlympiadBench,一个多模态基准,包含奥林匹克级别的数学和物理问题 🔥🔥🔥
  • [2024-11-16] 支持 WildVision,一个基于多模态竞技场数据的主观多模态基准 🔥🔥🔥
  • [2024-11-13] 支持 MIA-Bench,一个多模态指令跟随基准 🔥🔥🔥
  • [2024-11-08] 支持 Aria,一个多模态原生 MoE 模型,感谢 teowu 🔥🔥🔥
  • [2024-11-04] 支持 WorldMedQA-V,该基准包含 1000 多个医学 VQA 问题,涵盖巴西、以色列、日本、西班牙等四个国家的语言,以及它们的英文翻译 🔥🔥🔥
  • [2024-11-01] 支持 AUTO_SPLIT 标志 (https://github.com/open-compass/VLMEvalKit/pull/566),用于在低配置 GPU 上进行评估。设置后,模型将自动拆分到多个 GPU(流水线并行)以减少 GPU 内存使用(目前仅支持部分 VLMs:Qwen2-VL、Llama-3.2、LLaVA-OneVision 等) 🔥🔥🔥
  • [2024-10-30] 支持评估 MLVUTempCompass。这两个基准将很快被纳入 OpenVLM 视频排行榜 🔥🔥🔥

🏗️ 快速开始

请参阅快速开始获取入门指南。

📊 评测结果,支持的数据集和模型

评测结果

OpenVLM Leaderboard: 下载全部细粒度测试结果.

支持的图文多模态评测集

* 我们只提供了部分模型上的测试结果,剩余模型无法在 zero-shot 设定下测试出合理的精度

+ 我们尚未提供这个评测集的测试结果

- VLMEvalKit 仅支持这个评测集的推理,无法输出最终精度

$$^1$$ VLMEvalKit 在评测集的官方代码库中被使用

如果您设置了 API KEY,VLMEvalKit 将使用一个 LLM 从输出中提取答案进行匹配判断,否则它将使用精确匹配模式 (直接在输出字符串中查找“yes”,“no”,“A”,“B”,“C”等)。精确匹配只能应用于是或否任务和多选择任务

支持的视频多模态评测集

Dataset Dataset Names (for run.py) Task Dataset Dataset Names (for run.py) Task
MMBench-Video MMBench-Video VQA Video-MME Video-MME MCQ
MVBench MVBench/MVBench_MP4 MCQ MLVU MLVU MCQ & VQA
TempCompass TempCompass MCQ & Y/N & Caption

支持的模型

API 模型

GPT-4v (20231106, 20240409) 🎞️🚅 GPT-4o 🎞️🚅 Gemini-1.0-Pro 🎞️🚅 Gemini-1.5-Pro 🎞️🚅 Step-1V 🎞️🚅
Reka-[Edge / Flash / Core]🚅 Qwen-VL-[Plus / Max] 🎞️🚅
Qwen-VL-[Plus / Max]-0809 🎞️🚅
Claude3-[Haiku / Sonnet / Opus] 🎞️🚅 GLM-4v 🚅 CongRong 🎞️🚅
Claude3.5-Sonnet (20240620, 20241022) 🎞️🚅 GPT-4o-Mini 🎞️🚅 Yi-Vision🎞️🚅 Hunyuan-Vision🎞️🚅 BlueLM-V 🎞️🚅

基于 PyTorch / HF 的开源模型

IDEFICS-[9B/80B/v2-8B/v3-8B]-Instruct🚅🎞️ InstructBLIP-[7B/13B] LLaVA-[v1-7B/v1.5-7B/v1.5-13B] MiniGPT-4-[v1-7B/v1-13B/v2-7B]
mPLUG-Owl[2/3]🎞️ OpenFlamingo-v2🎞️ PandaGPT-13B Qwen-VL🚅🎞️
Qwen-VL-Chat🚅🎞️
VisualGLM-6B🚅 InternLM-XComposer-[1/2]🚅 ShareGPT4V-[7B/13B]🚅 TransCore-M
LLaVA (XTuner)🚅 CogVLM-[Chat/Llama3]🚅 ShareCaptioner🚅 CogVLM-Grounding-Generalist🚅
Monkey🚅
Monkey-Chat🚅
EMU2-Chat🚅🎞️ Yi-VL-[6B/34B] MMAlaya🚅
InternLM-XComposer-2.5🚅🎞️ MiniCPM-[V1/V2/V2.5/V2.6]🚅🎞️ OmniLMM-12B InternVL-Chat-[V1-1/V1-2/V1-5/V2]🚅🎞️
DeepSeek-VL🎞️ LLaVA-NeXT🚅🎞️ Bunny-Llama3🚅 XVERSE-V-13B
PaliGemma-3B 🚅 360VL-70B 🚅 Phi-3-Vision🚅🎞️
Phi-3.5-Vision🚅🎞️
WeMM🚅
GLM-4v-9B 🚅 Cambrian-[8B/13B/34B] LLaVA-Next-[Qwen-32B] 🎞️ Chameleon-[7B/30B]🚅🎞️
Video-LLaVA-7B-[HF] 🎬 VILA1.5-[3B/8B/13B/40B]🎞️ Ovis[1.5-Llama3-8B/1.5-Gemma2-9B/1.6-Gemma2-9B/1.6-Llama3.2-3B] 🚅🎞️ Mantis-8B-[siglip-llama3/clip-llama3/Idefics2/Fuyu] 🎞️
Llama-3-MixSenseV1_1🚅 Parrot-7B 🚅 OmChat-v2.0-13B-sinlge-beta 🚅 Video-ChatGPT 🎬
Chat-UniVi-7B[-v1.5] 🎬 LLaMA-VID-7B 🎬 VideoChat2-HD 🎬 PLLaVA-[7B/13B/34B] 🎬
RBDash_72b 🚅🎞️ xgen-mm-phi3-[interleave/dpo]-r-v1.5 🚅🎞️ Qwen2-VL-[2B/7B/72B]🚅🎞️ slime_[7b/8b/13b]🎞️
Eagle-X4-[8B/13B]🚅🎞️,
Eagle-X5-[7B/13B/34B]🚅🎞️
Moondream1🚅,
Moondream2🚅
XinYuan-VL-2B-Instruct🚅🎞️ Llama-3.2-[11B/90B]-Vision-Instruct🚅
Kosmos2🚅 H2OVL-Mississippi-[0.8B/2B]🚅🎞️ **Pixtral-12B**🎞️ **Falcon2-VLM-11B**🚅
**MiniMonkey**🚅🎞️ **LLaVA-OneVision**🚅🎞️ **LLaVA-Video**🚅🎞️ **Aquila-VL-2B**🚅🎞️
Mini-InternVL-Chat-[2B/4B]-V1-5🚅🎞️ InternVL2 Series 🚅🎞️ **Janus-1.3B**🚅🎞️ **molmoE-1B/molmo-7B/molmo-72B**🚅
**Points-[Yi-1.5-9B/Qwen-2.5-7B]**🚅 **NVLM**🚅 **VIntern**🚅🎞️ **Aria**🚅🎞️

🎞️ 表示支持多图片输入。

🚅 表示模型可以被直接使用,不需任何额外的配置。

🎬 表示支持视频输入。

其他

Transformers 的版本推荐:

请注意,某些 VLM 可能无法在某些特定的 transformers 版本下运行,我们建议使用以下设置来评估对应的VLM:

  • 请用 transformers==4.33.0 来运行: Qwen series, Monkey series, InternLM-XComposer Series, mPLUG-Owl2, OpenFlamingo v2, IDEFICS series, VisualGLM, MMAlaya, ShareCaptioner, MiniGPT-4 series, InstructBLIP series, PandaGPT, VXVERSE.
  • 请用 transformers==4.37.0 来运行: LLaVA series, ShareGPT4V series, TransCore-M, LLaVA (XTuner), CogVLM Series, EMU2 Series, Yi-VL Series, MiniCPM-[V1/V2], OmniLMM-12B, DeepSeek-VL series, InternVL series, Cambrian Series, VILA Series, Llama-3-MixSenseV1_1, Parrot-7B, PLLaVA Series.
  • 请用 transformers==4.40.0 来运行: IDEFICS2, Bunny-Llama3, MiniCPM-Llama3-V2.5, 360VL-70B, Phi-3-Vision, WeMM.
  • 请用 transformers==latest 来运行: LLaVA-Next series, PaliGemma-3B, Chameleon series, Video-LLaVA-7B-HF, Ovis series, Mantis series, MiniCPM-V2.6, OmChat-v2.0-13B-sinlge-beta, Idefics-3, GLM-4v-9B, VideoChat2-HD.

如何测试一个 VLM 是否可以正常运行:

from vlmeval.config import supported_VLM
model = supported_VLM['idefics_9b_instruct']()
# 前向单张图片
ret = model.generate(['assets/apple.jpg', 'What is in this image?'])
print(ret)  # 这张图片上有一个带叶子的红苹果
# 前向多张图片
ret = model.generate(['assets/apple.jpg', 'assets/apple.jpg', 'How many apples are there in the provided images? '])
print(ret)  # 提供的图片中有两个苹果

🛠️ 开发指南

要开发自定义评测数据集,支持其他 VLMs,或为 VLMEvalKit 贡献代码,请参阅开发指南

为激励来自社区的共享并分享相应的 credit,在下一次 report 更新中,我们将:

  • 致谢所有的 contribution
  • 具备三个或以上主要贡献 (支持新模型、评测集、或是主要特性) 的贡献者将可以加入技术报告的作者列表 。合条件的贡献者可以创建 issue 或是在 VLMEvalKit Discord Channel 私信 kennyutc,我们将进行跟进

🎯 VLMEvalKit 的目标

该代码库的设计目标是:

  1. 提供一个易于使用开源评估工具包,方便研究人员和开发人员评测现有的多模态大模型,并使评测结果易于复现
  2. 使 VLM 开发人员能够轻松地评测自己的模型。在多个支持的基准测试上评估 VLM,只需实现一个 generate_inner() 函数,所有其他工作负载(数据下载、数据预处理、预测推理、度量计算)都由代码库处理。

该代码库的设计目标不是:

复现所有第三方基准测试原始论文中报告的准确数字。有两个相关的原因:

  1. VLMEvalKit 对所有 VLMs 使用基于生成的评估(可选使用基于 LLM 的答案提取)。同时,一些基准测试可能官方使用不同的方法(例如,SEEDBench 使用基于 PPL 的评估)。对于这些基准测试,我们在相应的结果中比较两个得分。我们鼓励开发人员在代码库中支持其他评估范式。
  2. 默认情况下,我们对所有多模态模型使用相同的提示模板来评估基准测试。同时,一些多模态模型可能有他们特定的提示模板(目前可能未在代码库中涵盖)。我们鼓励 VLM 的开发人员在 VLMEvalKit 中实现自己的提示模板,如果目前未覆盖。这将有助于提高可复现性。

🖊️ 引用

如果我们的工作对您有所帮助,请考虑 star🌟 VLMEvalKit。感谢支持!

Stargazers repo roster for @open-compass/VLMEvalKit

如果您在研究中使用了 VLMEvalKit,或希望参考已发布的开源评估结果,请使用以下 BibTeX 条目以及与您使用的特定 VLM / 基准测试相对应的 BibTex 条目。

@misc{duan2024vlmevalkit,
      title={VLMEvalKit: An Open-Source Toolkit for Evaluating Large Multi-Modality Models},
      author={Haodong Duan and Junming Yang and Yuxuan Qiao and Xinyu Fang and Lin Chen and Yuan Liu and Xiaoyi Dong and Yuhang Zang and Pan Zhang and Jiaqi Wang and Dahua Lin and Kai Chen},
      year={2024},
      eprint={2407.11691},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2407.11691},
}

🔝回到顶部