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