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