• Stars
    star
    146
  • Rank 244,524 (Top 5 %)
  • Language
    Jupyter Notebook
  • License
    Apache License 2.0
  • Created over 1 year ago
  • Updated 3 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Perception Test: A Diagnostic Benchmark for Multimodal Video Models

News

Test Splits are now available for all the challenges! For Downloads see download section.

Join the first Perception Test challenge organised as an ICCV2023 workshop, website here ptchallenge-workshop.github.io.

Download data Download section below
Evaluation scripts (including data loader, dummy baseline, evaluation metrics) multiple-choice video QA, object tracking, action localisation, point tracking, sound localisation, grounded video QA
Evaluation server and leaderboard multiple-choice video QA, object tracking, action localisation, point tracking, sound localisation, grounded video QA

Overview

Perception Test: A Diagnostic Benchmark for Multimodal Video Models is a multimodal benchmark designed to comprehensively evaluate the perception and reasoning skills of multimodal video models. The Perception Test dataset introduces real-world videos designed to show perceptually interesting situations and defines multiple tasks (object and point tracking, action and sound localisation, multiple-choice and grounded video question-answering) that require understanding of memory, abstract patterns, physics, and semantics, across visual, audio, and text modalities.

In this repository, you will find:

  • A summary of the Perception Test and the associated challenge
  • A detailed description of the data and annotations in the Perception Test (interactive demo notebook here)
  • Details about how to download the data and annotations in the Perception Test (download section here)
  • Metrics for evaluating the performance on the different tasks (metrics section here)
  • Dummy baselines showcasing how to evaluate models on each of the tasks (baselines section here)

5-minutes summary of the Perception Test

Perception Test Overview Presentation

Try the Perception Test for yourself by accessing this quiz.

For more example videos in the Perception Test, check out this playlist.

Download the data and annotations

Test annotations will be released August 1st as part of the challenge phases.

The Perception Test dataset can be downloaded as zip files containing:

  • annotations in JSON format
  • videos (including audio) as MP4 files
  • audio-only files in WAV format
  • pre-computed features for the action localisation and sound localisation tasks.

Links

Task Split Videos Audio Labels
Sample All sample_videos.zip (214.9MB) sample_audios.zip (83.9MB) sample_annotations.zip (3MB)
All Tasks Train train_videos.zip (26.5GB) train_audios.zip (12.3GB) train_annotations.zip (30.6MB)
All Tasks Valid valid_videos.zip (70.2GB) valid_audios.zip (33.1GB) valid_annotations.zip (81.5MB)
All Tasks Test test_videos.zip (70.2GB) test_audios.zip (33.1GB) test_annotations.zip (81.5MB)

Challenge Downloads

Multi-Choice vQA
Challenge link: https://eval.ai/web/challenges/challenge-page/2091/overview

Task Split Videos Audio Labels
Multi-Choice vQA Train Use full split download above Use full split download above mc_question_train_annotations.zip (85kB)
Multi-Choice vQA Valid Use full split download above Use full split download above mc_question_valid_annotations.zip (200kB)
Multi-Choice vQA Test Use full split download above Use full split download above mc_question_test_annotations.zip (200kB)

Single Object Tracking
Challenge link: https://eval.ai/web/challenges/challenge-page/2094/overview

Task Split Videos Audio Labels
Single Object Tracking Train Use full split download above N/A Use full split download above
Single Object Tracking Valid sot_valid_videos_challenge2023.zip (11.6GB) N/A sot_valid_annotations_challenge2023.zip (9MB)

Temporal Action Localisation
Challenge link: https://eval.ai/web/challenges/challenge-page/2101/overview

Task Split Videos Audio Labels Video Features (TSP)
Temporal Action Localisation Train Use full split download above Use full split download above action_localisation_train_annotations.zip (217kB) action_localisation_train_video_features.zip (81.7MB)
Temporal Action Localisation Valid Use full split download above Use full split download above action_localisation_valid_annotations.zip (593kB) action_localisation_valid_video_features.zip (219.2MB)

Baselines

In this repo we provide dummy baselines to demonstrate how to load the data, evaluate and recreate some baseline results from the paper. For the other results in the baselines section in the paper, we will be adding another external repo.

Computational task Baseline
Object tracking Static baseline
Multiple-choice video QA Frequency dummy baseline
Point tracking Static baseline (available soon)

Metrics

Computational task Metric
Object tracking mean IoU
Point tracking Jaccard
Temporal action localisation mean Average Precision
Temporal sound localisation mean Average Precision
Multiple-choice video QA top-1 accuracy
Grounded video QA HOTA

Metrics code to evaluate performance for the different tasks coming soon.

Perception Test annotations

Explore the annotations: data_visualisation.ipynb

Summary

Annotation type Number of videos Number of annotations
Object tracks 11,609 189,940
Point tracks 145 8,647
Action segments 11,353 73,503
Sound segments 11,433 137,128
Multiple-choice video QA 10,361 38,060
Grounded video QA 3,063 6,086

Video metadata

Field Name Description
split The data split the video belongs to, one of ['train','valid','test'].
video_id The ID of the video ['video_xxxx'].
frame_rate The frame rate of the video in frames per second.
num_frames The total number of frames in the video.
resolution The height and width of the video in pixels.
audio_samples The total number of audio samples in the video.
audio_sample_rate The sample rate of the audio in the video in Hz.
is_cup_game Whether the video shows a cups-game or not, see paper for details.
is_camera_moving Whether the camera used to film the video is moving or not.

Object tracks

Field Name Description
id A unique annotation ID for each object track
label The name of the object, can also contain object attributes, e.g. red box.
is_occluder Whether the object occludes other objects in the video (This is valid only for the cups-games videos).
bounding_boxes The coordinates of the object's bounding box in the format [x1,y1,x2,y2] shape [n,4] where n is the number of annotated frames.
initial_tracking_box one-hot vector indicating which box annotation should be used to start the tracking for this object [n].
frame_ids The IDs of the frames that are annotated, normally 1 per second, e.g. 0, 30, 60, etc. shape [n].
timestamps The timestamps of the annotated frames in ms. shape [n].
is_masked Whether the object is masked in the annotated frame, corresponds to the bounding boxes [n] ( This is valid only for the cups-games videos).

Point tracks

Field Name Description
id A unique annotation ID for each point track.
label The label of the point track.
parent_objects The id of the object that the point belongs to.
frame_ids The IDs of the frames that are annotated, normally 0, 1, 2 etc. shape [N], where N is the total number of points in the track.
points The coordinates of the points in [y,x], shape [N, 2].

Action segments

Field Name Description
id A unique annotation ID for each action segment.
label The templated class of the action segment, e.g. Putting something into something.
parent_objects The ids of the objects involved in the action, can be empty, single, multiple or -1 for an object not annotated.
timestamps The start and end timestamps of the action segment in ms [start time, end time].
frame_ids The start and end frame IDs of the action segment [start frame, end frame].
label_id A unique class ID for each label in the dataset.

Sound segments

Field Name Description
id A unique annotation ID for each sound segment.
label The name or class of the sound segment.
parent_objects The object ids related to this sound segment, can be empty, single, multiple or -1 for an object not annotated.
timestamps The start and end timestamps of the sound segment in ms [start time, end time].
frame_ids The start and end frame IDs of the sound segment [start frame, end frame].
is_visible Whether the objects causing the sound in this segment are visible or not, e.g. if an object falls off the table and the impact point with the floor is occluded, then is_visible=False.
label_id A unique class ID for each label in the dataset.

Multiple-choice video question-answers

Field Name Description
id A unique annotation ID for each question.
question The text of the question.
options The possible options for the question. There are 3 possible options, and only one is correct.
answer_id The ID of the correct option for the question.
area The skill area the question pertains to. Can be Memory, Abstraction, Physics, Semantics.
reasoning The type of reasoning required to answer the question. Can be Descriptive, Explanatory, Predictive, or Counterfactual.
tag Different skills involved in answering the given question. A question can have multiple skill tags.

Grounded video question-answers

Field Name Description
id A unique annotation ID for each question.
question The text of the question.
answers The answer for the question given as a list of object ids.
area The skill area the question pertains to. Can be Memory, Abstraction, Physics, Semantics.
reasoning The type of reasoning required to answer the question. Can be Descriptive, Explanatory, Predictive, or Counterfactual.

Feedback and support

If you have any questions, feedback, or require support regarding the Perception Test dataset or challenge, please contact us at [email protected].

Citing this work

@misc{patraucean2023perception,
      title={Perception Test: A Diagnostic Benchmark for Multimodal Video Models}, 
      author={Viorica Pătrăucean and Lucas Smaira and Ankush Gupta and Adrià Recasens Continente and Larisa Markeeva and Dylan Banarse and Skanda Koppula and Joseph Heyward and Mateusz Malinowski and Yi Yang and Carl Doersch and Tatiana Matejovicova and Yury Sulsky and Antoine Miech and Alex Frechette and Hanna Klimczak and Raphael Koster and Junlin Zhang and Stephanie Winkler and Yusuf Aytar and Simon Osindero and Dima Damen and Andrew Zisserman and João Carreira},
      year={2023},
      eprint={2305.13786},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

License and disclaimer

Copyright 2022 DeepMind Technologies Limited

All software is licensed under the Apache License, Version 2.0 (Apache 2.0); you may not use this file except in compliance with the Apache 2.0 license. You may obtain a copy of the Apache 2.0 license at: https://www.apache.org/licenses/LICENSE-2.0

All other materials are licensed under the Creative Commons Attribution 4.0 International License (CC-BY). You may obtain a copy of the CC-BY license at: https://creativecommons.org/licenses/by/4.0/legalcode

Unless required by applicable law or agreed to in writing, all software and materials distributed here under the Apache 2.0 or CC-BY licenses are distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the licenses for the specific language governing permissions and limitations under those licenses.

This is not an official Google product.

More Repositories

1

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook
12,817
star
2

alphafold

Open source code for AlphaFold.
Python
11,700
star
3

sonnet

TensorFlow-based neural network library
Python
9,691
star
4

pysc2

StarCraft II Learning Environment
Python
7,904
star
5

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.
Jupyter Notebook
7,202
star
6

lab

A customisable 3D platform for agent-based AI research
C
7,012
star
7

graph_nets

Build Graph Nets in Tensorflow
Python
5,325
star
8

graphcast

Python
4,242
star
9

learning-to-learn

Learning to Learn in TensorFlow
Python
4,063
star
10

open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
C++
4,019
star
11

alphageometry

Python
3,580
star
12

dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
Python
3,473
star
13

acme

A library of reinforcement learning components and agents
Python
3,372
star
14

trfl

TensorFlow Reinforcement Learning
Python
3,139
star
15

dm-haiku

JAX-based neural network library
Python
2,697
star
16

alphatensor

Python
2,616
star
17

dnc

A TensorFlow implementation of the Differentiable Neural Computer.
Python
2,478
star
18

mctx

Monte Carlo tree search in JAX
Python
2,209
star
19

gemma

Open weights LLM from Google DeepMind.
Jupyter Notebook
2,061
star
20

code_contests

C++
2,010
star
21

kinetics-i3d

Convolutional neural network model for video classification trained on the Kinetics dataset.
Python
1,639
star
22

mathematics_dataset

This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.
Python
1,621
star
23

optax

Optax is a gradient processing and optimization library for JAX.
Python
1,492
star
24

bsuite

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
Python
1,465
star
25

penzai

A JAX research toolkit for building, editing, and visualizing neural networks.
Python
1,405
star
26

educational

Jupyter Notebook
1,382
star
27

jraph

A Graph Neural Network Library in Jax
Python
1,306
star
28

rc-data

Question answering dataset featured in "Teaching Machines to Read and Comprehend
Python
1,285
star
29

rlax

Python
1,185
star
30

tapnet

Tracking Any Point (TAP)
Python
1,033
star
31

scalable_agent

A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.
Python
972
star
32

neural-processes

This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).
Jupyter Notebook
966
star
33

android_env

RL research on Android devices.
Python
946
star
34

mujoco_menagerie

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
Jupyter Notebook
926
star
35

dramatron

Dramatron uses large language models to generate coherent scripts and screenplays.
Jupyter Notebook
904
star
36

tree

tree is a library for working with nested data structures
Python
891
star
37

xmanager

A platform for managing machine learning experiments
Python
796
star
38

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control
C++
771
star
39

materials_discovery

Python
770
star
40

chex

Python
716
star
41

reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
C++
692
star
42

alphadev

Python
662
star
43

pycolab

A highly-customisable gridworld game engine with some batteries included. Make your own gridworld games to test reinforcement learning agents!
Python
654
star
44

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Python
643
star
45

hanabi-learning-environment

hanabi_learning_environment is a research platform for Hanabi experiments.
Python
614
star
46

funsearch

Jupyter Notebook
611
star
47

ai-safety-gridworlds

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.
Python
577
star
48

dqn

Lua/Torch implementation of DQN (Nature, 2015)
Lua
546
star
49

ithaca

Restoring and attributing ancient texts using deep neural networks
Jupyter Notebook
540
star
50

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.
Python
516
star
51

distrax

Python
509
star
52

recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.
Python
505
star
53

surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.
Python
493
star
54

tracr

Python
467
star
55

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods
Jupyter Notebook
463
star
56

alphamissense

Python
455
star
57

narrativeqa

This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.
Shell
432
star
58

lab2d

A customisable 2D platform for agent-based AI research
C++
415
star
59

open_x_embodiment

Jupyter Notebook
409
star
60

dqn_zoo

DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (DQN) agent.
Python
406
star
61

clrs

Python
376
star
62

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment
Python
366
star
63

dm_pix

PIX is an image processing library in JAX, for JAX.
Python
363
star
64

concordia

A library for generative social simulation
Python
351
star
65

mathematics_conjectures

Jupyter Notebook
348
star
66

alphastar

Python
346
star
67

spiral

We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ images
C++
327
star
68

dm_env

A Python interface for reinforcement learning environments
Python
326
star
69

dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.
Python
315
star
70

uncertain_ground_truth

Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggregation from our recent work on uncertain ground truth (TMLR'23 and ArXiv pre-print).
Python
315
star
71

long-form-factuality

Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
Python
314
star
72

launchpad

Python
305
star
73

leo

Implementation of Meta-Learning with Latent Embedding Optimization
Python
302
star
74

streetlearn

A C++/Python implementation of the StreetLearn environment based on images from Street View, as well as a TensorFlow implementation of goal-driven navigation agents solving the task published in “Learning to Navigate in Cities Without a Map”, NeurIPS 2018
C++
279
star
75

gqn-datasets

Datasets used to train Generative Query Networks (GQNs) in the ‘Neural Scene Representation and Rendering’ paper.
Python
269
star
76

enn

Python
265
star
77

multi_object_datasets

Multi-object image datasets with ground-truth segmentation masks and generative factors.
Python
247
star
78

AQuA

A algebraic word problem dataset, with multiple choice questions annotated with rationales.
238
star
79

card2code

A code generation dataset for generating the code that implements Hearthstone and Magic The Gathering card effects.
236
star
80

grid-cells

Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial agents, as published at https://www.nature.com/articles/s41586-018-0102-6
Python
236
star
81

arnheim

Jupyter Notebook
235
star
82

synjax

Python
233
star
83

torch-hdf5

Torch interface to HDF5 library
Lua
231
star
84

dm_memorytasks

A set of 13 diverse machine-learning tasks that require memory to solve.
Python
220
star
85

Temporal-3D-Pose-Kinetics

Exploiting temporal context for 3D human pose estimation in the wild: 3D poses for the Kinetics dataset
Python
214
star
86

opro

official code for "Large Language Models as Optimizers"
Python
199
star
87

dm_alchemy

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark
Python
197
star
88

neural_testbed

Jupyter Notebook
187
star
89

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Python
187
star
90

pg19

179
star
91

xquad

173
star
92

jmp

JMP is a Mixed Precision library for JAX.
Python
171
star
93

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019
Python
165
star
94

abstract-reasoning-matrices

Progressive matrices dataset, as described in: Measuring abstract reasoning in neural networks (Barrett*, Hill*, Santoro*, Morcos, Lillicrap), ICML2018
162
star
95

neural_networks_chomsky_hierarchy

Neural Networks and the Chomsky Hierarchy
Python
162
star
96

xitari

This is the 0.4 release of the Arcade Learning Environment (ALE), a platform designed for AI research. ALE is based on Stella, an Atari 2600 VCS emulator.
C++
159
star
97

tensor_annotations

Annotating tensor shapes using Python types
Python
158
star
98

symplectic-gradient-adjustment

A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"
Jupyter Notebook
150
star
99

mc_gradients

Jupyter Notebook
149
star
100

interval-bound-propagation

This repository contains a simple implementation of Interval Bound Propagation (IBP) using TensorFlow: https://arxiv.org/abs/1810.12715
Python
148
star