Prepare the info file for sunrgbd dataset. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info filenames. out_dir (str): Output directory of the generated info file. workers (int): Number of threads to be used.
(root_path, info_prefix, out_dir, workers)
| 121 | |
| 122 | |
| 123 | def sunrgbd_data_prep(root_path, info_prefix, out_dir, workers): |
| 124 | """Prepare the info file for sunrgbd dataset. |
| 125 | |
| 126 | Args: |
| 127 | root_path (str): Path of dataset root. |
| 128 | info_prefix (str): The prefix of info filenames. |
| 129 | out_dir (str): Output directory of the generated info file. |
| 130 | workers (int): Number of threads to be used. |
| 131 | """ |
| 132 | indoor.create_indoor_info_file( |
| 133 | root_path, info_prefix, out_dir, workers=workers) |
| 134 | |
| 135 | |
| 136 | def waymo_data_prep(root_path, |