Some files have wrong hardcoded paths during frame extraction

[skymanaditya1]

When I extract the frames on the hm3d dataset using - python data/hm3d/extract-frames.py, the code throws an error when processing the folder data/frames/hm3d-v0/021-hm3d-LT9Jq6dN3Ea in the val dataset.

The error is -
AssertionError: ESP_CHECK failed: No Stage Attributes exists for requested scene '/datasets01/hm3d/090121/val/00862-LT9Jq6dN3Ea/LT9Jq6dN3Ea.basis.glb' in currently specified Scene Dataset default. Likely the Scene Dataset Configuration requested was not found and so a new, empty Scene Dataset was created. Verify the Scene Dataset Configuration file name used.

This is because the .pkl being read has the path '/datasets01/hm3d/090121/...' being hardcoded into it when its read as data['scene_id']. I added a temporary fix, but it needs to be looked into, otherwise the extract-frames.py fails atleast for this folder.

[zhuangaoooo]

I have met the same problem. Hope this bug will be solved soon.
My temporary solution:
Replace following functions in extract-frames.py

OLD_HM3D_ROOT = "/datasets01/hm3d/090121/val"

...

def replace_scene_path(original_path: str, new_root: Path) -> str:
    scene_id = Path(original_path).parent.name
    base_name = Path(original_path).name
    scene_dir = new_root / scene_id
    new_path = scene_dir / base_name
    return str(new_path)


def load_sim(path: Path, args: argparse.Namespace) -> habitat_sim.Simulator:
    data = pickle.load(path.open("rb"))
    if data["scene_id"].startswith(OLD_HM3D_ROOT):
        data["scene_id"] = replace_scene_path(data["scene_id"], args.hm3d_root)
    scene_id = data["scene_id"]
    agent_state = data["agent_state"]
    sensor_position = (
        agent_state.sensor_states["rgb"].position[1] - agent_state.position[1]
    )
    cfg = get_config(scene_id=scene_id, sensor_position=sensor_position)
    return habitat_sim.Simulator(cfg)

...

def extract_frames(folder: Path, args: argparse.Namespace) -> None:
    print("Extracting frames to: {}".format(folder))
    files = sorted(folder.glob("*.pkl"))
    sim = load_sim(path=files[0], args=args)

    print("Processing {} agent positions...".format(len(files)))
    for idx, path in enumerate(files):
        # set agent state
        data = pickle.load(path.open("rb"))
        agent = sim.get_agent(0)
        agent.set_state(data["agent_state"])

        # save data
        if not args.rgb_only:
            if idx == 0:
                save_intrinsics(path)
            save_pose(path, sim)
            save_depth(path, sim)
        save_color(path, sim)

    sim.close()
    print("Extracting frames to: {} done!".format(folder))