遇见数据集

j-min/layoutbench-coco

收藏
Hugging Face2024-07-21 更新2024-07-22 收录
官方服务:

资源简介:

LayoutBench-COCO是一个诊断基准,用于评估布局引导的图像生成模型在任意未见过的布局上的表现。与LayoutBench不同,LayoutBench-COCO包含真实对象的OOD布局,并支持零样本评估。数据集测量了四个技能(数量、位置、大小、组合),这些技能的对象来自MS COCO数据集。新的‘组合’拆分包含具有不同空间关系的两个对象的布局,其余三个拆分与LayoutBench类似。

LayoutBench-COCO is a diagnostic benchmark dataset for evaluating layout-guided image generation models on arbitrary, unseen layouts. The dataset consists of OOD layouts of real objects from MS COCO and supports zero-shot evaluation. LayoutBench-COCO measures four skills (Number, Position, Size, Combination), with the combination skill including layouts of two objects in different spatial relations. The dataset is divided into multiple config files, each containing different types of layout data.

提供机构:
j-min
原始信息汇总

LayoutBench-COCO 数据集概述

基本信息

  • 许可证: MIT
  • 任务类别: 文本到图像生成
  • 语言: 英语
  • 配置:
    • 默认配置:
      • 分割: combinations_common
        • 路径: combinations_layouts_common.json
      • 分割: combinations_uncommon
        • 路径: combinations_layouts_uncommon.json
      • 分割: count_2_4
        • 路径: count_layouts_2_4.json
      • 分割: count_5_7
        • 路径: count_layouts_5_7.json
      • 分割: count_8_10
        • 路径: count_layouts_8_10.json
      • 分割: position_boundary
        • 路径: position_layouts_boundary.json
      • 分割: position_center
        • 路径: position_layouts_center.json
      • 分割: size_large
        • 路径: size_layouts_large.json
      • 分割: size_small
        • 路径: size_layouts_small.json
  • 数据集名称: LayoutBench-COCO

数据集描述

数据集特点

  • 目标: 评估布局引导的图像生成模型在任意、未见过的布局上的表现。
  • 区别: 与LayoutBench不同,LayoutBench-COCO包含真实对象的OOD布局,并支持零样本评估。
  • 技能测量: 包含四个技能(数量、位置、大小、组合),对象来自MS COCO。

技能详情

技能1: 数量

  • 定义: 为2到10个对象定义两种布局,使用40个COCO对象,共720个布局(= 2×9×40)。
  • 分割: 2到4个对象的布局称为“few”分割,8到10个对象的布局称为“many”分割。
  • 描述: 布局与模板为“a photo of [N] [objects]”的标题配对。

技能2: 位置

  • 定义: 为边界和中心分割分别定义四种布局,使用40个COCO对象,共320个布局(= 2 × 4 × 40)。
  • 描述: 布局与模板为“a photo of [N] [objects]”的标题配对。

技能3: 大小

  • 定义: 为tiny和large分割分别定义九种布局,使用40个COCO对象,共720个布局(= 9×2×40)。
  • 描述: 布局与模板为“a photo of [N] [objects]”的标题配对。

技能4: 组合

  • 定义: 测量模型是否能生成在现实世界中常见或不常见的两个对象。
  • 布局: 为三种空间关系(holding, next to, sitting on)定义三种布局,不指定对象。
  • 对象对: 为每种关系手动定义20对COCO对象,分为常见和不常见分割。
  • 示例: 例如,“person sitting on chair”比“elephant sitting on banana”更常见。
  • 总数: 共360个布局(= 2×3×3×20)。
  • 描述: 布局与模板为“[objA] [relation] [objB]”的标题配对。

使用指南

  • 步骤1: 下载LayoutBench-COCO并从布局生成图像。
  • 步骤2: 运行评估。

引用

bibtex @inproceedings{Cho2024LayoutBench, author = {Jaemin Cho and Linjie Li and Zhengyuan Yang and Zhe Gan and Lijuan Wang and Mohit Bansal}, title = {Diagnostic Benchmark and Iterative Inpainting for Layout-Guided Image Generation}, booktitle = {The First Workshop on the Evaluation of Generative Foundation Models}, year = {2024}, }

搜集汇总
数据集介绍
j-min/layoutbench-coco 数据集图片
构建方式
LayoutBench-COCO数据集源自CVPR 2024 Workshop的布局引导图像生成诊断基准,其构建基于对MS COCO真实对象的系统化布局设计。数据集围绕四项核心技能——数量、位置、尺寸与组合——分别生成多样化布局:数量技能涵盖2至10个对象的布局,位置技能定义边界与中心两种空间配置,尺寸技能区分大与小两种尺度,而组合技能则通过常见与罕见空间关系(如“持有”、“相邻”、“坐于”)配对对象,形成共计2120个布局样本。每个布局均配以模板化文本描述,确保评估的标准化与可重复性。
使用方法
使用LayoutBench-COCO需遵循两步流程:首先从HuggingFace下载数据集,并依据提供的布局JSON文件生成对应图像,具体步骤参考项目仓库中的图像生成指南;随后利用YOLOv7等检测工具对生成图像进行自动评估,计算布局一致性指标。数据集以多种配置分片形式提供,用户可根据研究需求选择特定技能子集进行测试。评估代码与详细说明均托管于GitHub,便于复现基准结果并与现有布局引导生成模型进行公平比较。
背景与挑战
背景概述
布局引导的图像生成是文本到图像合成领域的前沿方向,旨在根据用户指定的物体空间位置关系生成符合预期的视觉内容。然而,现有模型在面对未见过的布局组合时往往表现不佳,暴露出泛化能力的严重不足。在此背景下,来自微软研究院、北卡罗来纳大学教堂山分校等机构的研究人员Jaemin Cho、Linjie Li、Zhengyuan Yang等人于2024年CVPR研讨会上发布了LayoutBench-COCO数据集。该数据集专注于评估模型在真实物体(源自MS COCO)的开放域布局上的零样本生成能力,核心研究问题在于衡量模型在数量、位置、大小和组合四种技能上的表现。其中,组合技能特别考察了模型对常见与罕见空间关系的区分能力,为诊断布局引导生成模型的鲁棒性提供了标准化基准,对推动该领域评估体系的完善产生了重要影响。
当前挑战
LayoutBench-COCO所解决的核心领域挑战在于现有布局引导生成模型对任意、未见布局的泛化能力薄弱,具体体现在四个方面:数量技能中模型难以准确生成指定数量的物体(如2-10个);位置技能要求物体严格位于边界或中心区域;大小技能需区分微小与大型物体的尺度差异;组合技能则考验模型对常见(如人坐在椅子上)与罕见(如大象坐在香蕉上)空间关系的辨别力。在数据集构建过程中,挑战同样显著:需从COCO中精选40个物体类别,并为每种技能手工设计多样化的布局模板,同时为组合技能人工标注20对常见与罕见物体对及其三种空间关系,确保布局的科学性与诊断有效性。此外,生成与评估流程需整合YOLOv7等检测器,以自动化验证生成图像中物体属性是否符合布局约束,构建了从布局输入到性能度量的完整闭环。
常用场景
经典使用场景
LayoutBench-COCO数据集专为评估布局引导的图像生成模型而设计,其经典使用场景在于对模型在任意未见布局上的生成能力进行诊断性测试。通过涵盖数量、位置、大小和组合四种核心技能,该数据集能够系统性地检验模型在处理复杂空间关系时的鲁棒性与泛化能力。例如,在组合技能中,数据集通过设计常见与罕见物体对(如“人坐在椅子上”与“大象坐在香蕉上”),衡量模型对现实世界物体共现规律的把握程度。这种细粒度的评估范式为对比不同生成架构(如扩散模型与自回归模型)在布局忠实度上的表现提供了标准化基准。
解决学术问题
该数据集有效解决了布局引导图像生成领域中缺乏系统性诊断基准的学术难题。现有评估指标多聚焦于图像整体质量,却忽视了模型对布局约束的精确遵循程度。LayoutBench-COCO通过解耦数量、位置、大小和组合四个维度,能够精确定位模型在特定技能上的缺陷,例如模型在生成包含大量物体(8-10个)或物体位于边界位置时的失败模式。这一设计推动了从“整体质量评估”向“技能分解评估”的范式转变,为理解生成模型的认知边界提供了方法论支撑,并启发了后续研究对布局条件建模的优化方向。
实际应用
在实际应用中,LayoutBench-COCO可服务于需要高精度空间控制的图像生成场景,如虚拟场景构建、广告创意设计及辅助视觉创作工具。例如,在电商产品展示中,设计师可能需要生成“一个咖啡杯放在笔记本电脑右侧”的特定布局,而该数据集训练的评估框架能确保生成模型严格遵循此类空间指令。此外,在自动驾驶仿真中,通过指定“五辆车排列在十字路口”的布局,该数据集可检验生成模型对复杂交通场景的再现能力,从而降低对真实数据的依赖。其零样本评估特性还支持在未见布局上的即时测试,提升了部署的灵活性。
数据集最近研究
最新研究方向
LayoutBench-COCO数据集聚焦于布局引导图像生成模型的诊断性评估,尤其在处理未见过的、分布外(OOD)布局时展现出独特价值。该数据集围绕数量、位置、尺寸和组合四大核心技能设计,其中“组合”技能通过评估模型生成常见与罕见空间关系对象对的能力,直击现实世界图像合成中的语义合理性挑战。这一研究方向与当前生成式基础模型在可控性、泛化性方面的前沿探索紧密相连,其提出的迭代式修复方法为改善布局约束下的图像生成质量提供了新范式,对推动多模态生成模型在复杂场景下的鲁棒性评估与优化具有重要影响。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务