train_classification.py
把第91行修改为自己的路径,就可以运行了
test_cla.py,需要训练完才能运行测试,我没训练完,所以报错显示我没有best.pth文件
网盘里面是我运行的训练和测试的视频,以及源代码,数据我没上传完,数据直接点击官网的链接下载就行。
链接:https://pan.baidu.com/s/1vMVdf6pNbr6hH7IzU5kwaA?pwd=1234
提取码:1234
--来自百度网盘超级会员V4的分享
我是把train_par和test_par跑了一下,把train_par的epoch设置成2,然后train,生成best.pth,运行test_par,其中,如果numpy的版本高于1.23会报错,所以需要将train和test代码中的153行,这个部分,修改为 dtype=float,然后就可以完美运行了,运行的视频在百度网盘里面。