| 56 | return lmdb_sets |
| 57 | |
| 58 | def dataset_traversal(self): |
| 59 | lmdb_num = len(self.lmdb_sets) |
| 60 | total_sample_num = 0 |
| 61 | for lno in range(lmdb_num): |
| 62 | total_sample_num += self.lmdb_sets[lno]['num_samples'] |
| 63 | data_idx_order_list = np.zeros((total_sample_num, 2)) |
| 64 | beg_idx = 0 |
| 65 | for lno in range(lmdb_num): |
| 66 | tmp_sample_num = self.lmdb_sets[lno]['num_samples'] |
| 67 | end_idx = beg_idx + tmp_sample_num |
| 68 | data_idx_order_list[beg_idx:end_idx, 0] = lno |
| 69 | data_idx_order_list[beg_idx:end_idx, |
| 70 | 1] = list(range(tmp_sample_num)) |
| 71 | data_idx_order_list[beg_idx:end_idx, 1] += 1 |
| 72 | beg_idx = beg_idx + tmp_sample_num |
| 73 | return data_idx_order_list |
| 74 | |
| 75 | def get_ext_data(self): |
| 76 | ext_data_num = 0 |