| import re | |
| LABEL_PATTERN = re.compile(r'\s*(\d+)(.+)') | |
| def load_labels(path): | |
| with open(path, 'r', encoding='utf-8') as f: | |
| lines = (LABEL_PATTERN.match(line).groups() for line in f.readlines()) | |
| return {int(num): text.strip() for num, text in lines} | |
| def input_image_size(engine): | |
| _, h, w, _ = engine.get_input_tensor_shape() | |
| return w, h | |
| def same_input_image_sizes(engines): | |
| return len({input_image_size(engine) for engine in engines}) == 1 |