make_pointnav_debug_dataset.py

import os

from constants import ABS_PATH_OF_TOP_LEVEL_DIR
from plugins.robothor_plugin.scripts.make_objectnav_debug_dataset import (
    create_debug_dataset_from_train_dataset,
)

if __name__ == "__main__":
    SCENE = "FloorPlan1"
    EPISODES = [0, 7, 11, 12]
    BASE_OUT = os.path.join(
        ABS_PATH_OF_TOP_LEVEL_DIR, "datasets", "ithor-pointnav", "debug"
    )

    create_debug_dataset_from_train_dataset(
        scene=SCENE,
        target_object_type=None,
        episodes_subset=EPISODES,
        train_dataset_path=os.path.join(
            ABS_PATH_OF_TOP_LEVEL_DIR, "datasets", "ithor-pointnav", "train"
        ),
        base_debug_output_path=BASE_OUT,
    )