(class_file)
| 116 | |
| 117 | |
| 118 | def load_classes(class_file): |
| 119 | with open(class_file) as f: |
| 120 | classes = [] |
| 121 | for line in f: |
| 122 | line = line.strip().split('\t') |
| 123 | if len(line) != 2: |
| 124 | continue |
| 125 | gate, class_name = int(line[0]), line[1] |
| 126 | if gate > 0: |
| 127 | classes.append(class_name) |
| 128 | classes = tuple(classes) |
| 129 | num_classes = len(classes) |
| 130 | logging.info('class number is {}'.format(num_classes)) |
| 131 | return classes, num_classes |
| 132 | |
| 133 | |
| 134 | def load_checkpoint(checkpoint, obj_model, classifier): |