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.
class Dataset
Base class for dataset specification.
class Episode
Base class for episode specification that includes initial position and rotation of agent, scene id, episode.
class EpisodeIterator
Episode Iterator class that gives options for how a list of episodes should be iterated.

Data

ALL_SCENES_MASK = '*'
T = ~T
TYPE_CHECKING = False