在构建resnet等机器识别神经网络时出现预测准确度高但类别不对的情况

ImageFolder 自动生成的 class_names 可能和训练时的 class_names 顺序不同，导致 predicted.item() 对应的类别索引错误。

解决方案：

• 在训练时保存类别顺序：

  import json with open("class_names.json", "w") as f:
      json.dump(dataset.classes, f)

• 在推理时加载：

  with open("class_names.json", "r") as f: 
      class_names = json.load(f)