habitat.core.dataset module

Implements dataset functionality to be used habitat.EmbodiedTask. habitat.core.dataset abstracts over a collection of habitat.core.Episode. Each episode consists of a single instantiation of a habitat.Agent inside habitat.Env.

Classes

class BaseEpisode
Base class for episode specification that includes only the episode_id and scene id. This class allows passing the minimum required episode information to identify the episode (unique key) to the habitat baseline process, thus saving evaluation time. :property episode_id: id of episode in the dataset, usually episode number. :property scene_id: id of scene in dataset.
class Dataset
Base class for dataset specification.
class Episode
class EpisodeIterator
Episode Iterator class that gives options for how a list of episodes should be iterated.

Data

ALL_SCENES_MASK = '*'
T = ~T