BEHAVIOR Vision Suite: Customizable Dataset Generation via Simulation
Authors: name: Yunhao Ge; name: Yihe Tang; name: Jiashu Xu; name: Cem Gokmen; name: Chengshu Li; name: Wensi Ai; name: Benjamin Jose Martinez; name: Arman Aydin; name: Mona Anvari; name: Ayush K Chakravarthy; name: Hong-Xing Yu; name: Josiah Wong; name: Sanjana Srivastava; name: Sharon Lee; name: Shengxin Zha; name: Laurent Itti; name: Yunzhu Li; name: Roberto Martín-Martín; name: Miao Liu; name: Pengchuan Zhang; name: Ruohan Zhang; name: Li Fei-Fei; name: Jiajun Wu
Published: 2024-05-15 17:57:56

Abstract: The systematic evaluation and understanding of computer vision models under
varying conditions require large amounts of data with comprehensive and
customized labels, which real-world vision datasets rarely satisfy. While
current synthetic data generators offer a promising alternative, particularly
for embodied AI tasks, they often fall short for computer vision tasks due to
low asset and rendering quality, limited diversity, and unrealistic physical
properties. We introduce the BEHAVIOR Vision Suite (BVS), a set of tools and
assets to generate fully customized synthetic data for systematic evaluation of
computer vision models, based on the newly developed embodied AI benchmark,
BEHAVIOR-1K. BVS supports a large number of adjustable parameters at the scene
level (e.g., lighting, object placement), the object level (e.g., joint
configuration, attributes such as "filled" and "folded"), and the camera level
(e.g., field of view, focal length). Researchers can arbitrarily vary these
parameters during data generation to perform controlled experiments. We
showcase three example application scenarios: systematically evaluating the
robustness of models across different continuous axes of domain shift,
evaluating scene understanding models on the same set of images, and training
and evaluating simulation-to-real transfer for a novel vision task: unary and
binary state prediction. Project website:
https://behavior-vision-suite.github.io/