Get alphabet and label names, union across all dirs.
(dirs)
| 144 | |
| 145 | |
| 146 | def _get_names(dirs): |
| 147 | """Get alphabet and label names, union across all dirs.""" |
| 148 | alphabets = set() |
| 149 | label_names = {} |
| 150 | for d in dirs: |
| 151 | for example in _walk_omniglot_dir(d): |
| 152 | alphabet, alphabet_char_id, label, _, _ = example |
| 153 | alphabets.add(alphabet) |
| 154 | label_name = "%s_%d" % (alphabet, alphabet_char_id) |
| 155 | if label in label_names: |
| 156 | assert label_names[label] == label_name |
| 157 | else: |
| 158 | label_names[label] = label_name |
| 159 | label_names = [label_names[k] for k in sorted(label_names)] |
| 160 | return alphabets, label_names |
no test coverage detected