weixin_51597565的博客

K近邻算法 图像分类

AlexNet图像分类

基于UCM数据集的图像分类 GoogLeNet

2.加载数据集并训练，训练集计算损失值loss，测试集计算accuracy，保存训练好的网络参数-------------------------train.py。3.利用训练好的网络参数后，用自己找的图像进行分类-------------------------------------------predict.py。定义LeNet网络模型---------------model.py。我的解决办法是：在代码首行加入下面两行代码，希望能帮到你。注意：训练路径和测试路径需要改为你自己的路径。