json_classes_labels = json.dumps(train_data.class_to_idx) print() print("categories labels below:") print(json_classes_labels) # print classes with labels in json format print() # save categories labels to json file with open("../dist/labels.json", "w+") as f: f.write(json_classes_labels)

这段代码将训练集中的类别和对应的标签转换为json格式，并输出到控制台。接着将json格式的类别标签保存到文件中，方便后续的预测和展示。其中，class_to_idx是一个字典类型，表示类别名和对应的标签。利用json.dumps函数可以将字典类型转换为json格式字符串，方便输出和保存。最后，将json格式的类别标签保存到文件中。