Skip to content

Benchmark data and code for Question-Answering on Movie stories

Notifications You must be signed in to change notification settings

kibeomKim/MovieQA_benchmark

 
 

Repository files navigation

MovieQA

MovieQA: Understanding Stories in Movies through Question-Answering
Makarand Tapaswi, Yukun Zhu, Rainer Stiefelhagen, Antonio Torralba, Raquel Urtasun, and Sanja Fidler
IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, June 2016.
Project page | arXiv preprint | Read the paper | Explore the data


Benchmark Data

The data is made available in simple JSON / text files for easy access in any environment. We provide Python scripts to help you get started by providing simple data loaders.

To obtain access to the stories, and evaluate new approaches on the test data, please register at our benchmark website.

Python data loader

import MovieQA
mqa = MovieQA.DataLoader()

Explore

Movies are indexed using their corresponding IMDb keys. For example
mqa.pprint_movie(mqa.movies_map['tt0133093'])

QAs are stored as a standard Python list mqa.pprint_qa(mqa.qa_list[0])

Use

To get train or test splits of the QA along with a particular story, use
story, qa = mqa.get_story_qa_data('train', 'plot')

Currently supported story forms are: plot, split plot


Requirements

  • numpy
  • pysrt

About

Benchmark data and code for Question-Answering on Movie stories

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%