| import shutil | |
| if __name__ == "__main__": | |
| print("Convert model labels to human readable") | |
| source_file = 'config.json' | |
| old_file = 'config.json.bak' | |
| shutil.copyfile(source_file, old_file) | |
| LABEL_TO_CHAR = { | |
| "LABEL_0" : "0", | |
| "LABEL_1" : ".", | |
| "LABEL_2" : ",", | |
| "LABEL_3" : "?", | |
| "LABEL_4" : "-", | |
| "LABEL_5" : ":" | |
| } | |
| with open(old_file, 'r') as fp_input, open(source_file, 'w') as fp_output: | |
| lines = fp_input.readlines() | |
| for line in lines: | |
| for label in LABEL_TO_CHAR.keys(): | |
| line = line.replace(label, LABEL_TO_CHAR[label]) | |
| fp_output.write(line) | |