Skip to content

Latest commit

 

History

History
11 lines (8 loc) · 994 Bytes

README.md

File metadata and controls

11 lines (8 loc) · 994 Bytes

The ArtObjSim dataset can be found here.

The train / val / test split can be found in the train_val_test_split.npy file in this repo. The usage is as follows:

>>> import numpy as np
>>> train_val_test_split = np.load('train_val_test_split.npy', allow_pickle=True).item()
>>> train_val_test_split['train']

Code for generating the dataset has also been provided. scenes_dict.npy contains the kitchen locations and viewing angles for each of the HM3D scenes used. Images can be rendered out from these viewing locations and orientations using the Habitat simulator. The raw scale annotations for the resulting images can be found under scale_annotations. Finally, to run the processing script to generate the full dataset, run src/post_processing.py with a path to the scale annotation json file (--json_filename) and the path to the kitchen locations dictionary (--path_to_scenes_dict).