| 201 | return opencv_c2w.tolist() |
| 202 | |
| 203 | def get_data_paths( |
| 204 | self, |
| 205 | jsonl_path_list, |
| 206 | data_dir_list, |
| 207 | num_used_data, |
| 208 | shuffle_lines, |
| 209 | shuffle_seed, |
| 210 | ): |
| 211 | data_paths = [] |
| 212 | for jsonl_path, image_dir, num_data_point in zip( |
| 213 | jsonl_path_list, data_dir_list, num_used_data |
| 214 | ): |
| 215 | with open(jsonl_path, 'r') as f: |
| 216 | raw_data = f.readlines() |
| 217 | if shuffle_lines: |
| 218 | self.rng.seed(shuffle_seed) |
| 219 | self.rng.shuffle(raw_data) |
| 220 | raw_data = raw_data[:num_data_point] |
| 221 | data_paths.extend([(json_data, image_dir) for json_data in raw_data]) |
| 222 | return data_paths |
| 223 | |
| 224 | def change_format(self, data, num_images): |
| 225 | elements = [] |