qq_53870613的博客

需要先配置好pytorch环境。 根据需要可更改测试集训练集的路径，具体方法可自行GPT。 test文件夹中存的测试集图片文件名不带标签，只能用于可视化模型效果，具体见代码原始test.py；ntest中是带标签的，可以统计正确率，具体见代码test.py。 原代码是在github上找的，本人重新改写了部分内容，主要是整合了数据集和测试集文件，并重新编写了部分代码，可以统计测试集的准确率。