(self, idx)
| 123 | return imgbuf, label |
| 124 | |
| 125 | def __getitem__(self, idx): |
| 126 | lmdb_idx, file_idx = self.data_idx_order_list[idx] |
| 127 | lmdb_idx = int(lmdb_idx) |
| 128 | file_idx = int(file_idx) |
| 129 | sample_info = self.get_lmdb_sample_info( |
| 130 | self.lmdb_sets[lmdb_idx]['txn'], file_idx) |
| 131 | if sample_info is None: |
| 132 | return self.__getitem__(np.random.randint(self.__len__())) |
| 133 | img, label = sample_info |
| 134 | data = {'image': img, 'label': label} |
| 135 | data['ext_data'] = self.get_ext_data() |
| 136 | outs = transform(data, self.ops) |
| 137 | if outs is None: |
| 138 | return self.__getitem__(np.random.randint(self.__len__())) |
| 139 | return outs |
| 140 | |
| 141 | def __len__(self): |
| 142 | return self.data_idx_order_list.shape[0] |
nothing calls this directly
no test coverage detected