| 103 | return lmdb_sets |
| 104 | |
| 105 | def dataset_traversal(self): |
| 106 | lmdb_num = len(self.lmdb_sets) |
| 107 | total_sample_num = 0 |
| 108 | for lno in range(lmdb_num): |
| 109 | total_sample_num += self.lmdb_sets[lno]['ratio_num_samples'] |
| 110 | data_idx_order_list = np.zeros((total_sample_num, 2)) |
| 111 | beg_idx = 0 |
| 112 | for lno in range(lmdb_num): |
| 113 | tmp_sample_num = self.lmdb_sets[lno]['ratio_num_samples'] |
| 114 | end_idx = beg_idx + tmp_sample_num |
| 115 | data_idx_order_list[beg_idx:end_idx, 0] = lno |
| 116 | data_idx_order_list[beg_idx:end_idx, 1] = list( |
| 117 | random.sample(range(1, self.lmdb_sets[lno]['num_samples'] + 1), |
| 118 | self.lmdb_sets[lno]['ratio_num_samples'])) |
| 119 | beg_idx = beg_idx + tmp_sample_num |
| 120 | return data_idx_order_list |
| 121 | |
| 122 | def get_img_data(self, value): |
| 123 | """get_img_data.""" |