qq_44644355的博客

前面我们训练的用的数据是ants(蚂蚁)和bees(蜜蜂), 所以预测时用的图片应该跟训练时类似。本次测试的图片为ants，如下图所示：代码中的图片路径和模型路径根据自己的情况来修改预测代码为：import torch, glob, cv2import numpy as npimport torch.nn as nnimport torch.nn.functional as Ffrom torchvision import models, transformsimport matplot

学习时不应该只关注算法模型的搭建，更应该关注怎么读取数据。代码是使用datasets.ImageFolder来读取图片数据的，后续要自己学习怎么将图片数据写入txt,然后通过txt文件来获取我们需要的数据1.代码知识对模型进行训练，并在测试集上检验模型的准确率。并没有用训练好的模型来预测一张图片的代码，后续需要添加。import torchfrom torch.autograd import Variableimport torchvisionfrom torchvision import da