Pytorch深度学习之实现病虫害图像分类

目录

一、pytorch框架1.1、概念1.2、机器学institute习与深度学习的区别1.3、在python中导入pytorch成功截图二、数据集三、代码复现3.1、导入第三方库3.2、cnn代码3.3、测试代码四、训练结果4.1、loss损失函数4.2、 acc4.3、单张图片识别准确率四、小结

一、pytorch框架

1.1、概念

pytorch是一个开源的python机器学习库，基于torch，用于自然语言处理等应用程序。

2017年1月，由facebook人工智能研究院（fair）基于torch推出了pytorch。它是一个基于python的可续计算包，提供两个高级功能：

1、具有强大的gpu加速的张量计算（如numpy）。

2、包含自动求导系统的深度神经网经典短句络。

1.2、机器学习与深度学习的区别

两者之间区别很多，在本篇博客中只简单描述一部分。以图片的形式展现。

前者为机器学习的过程。

后者为深度学习的过程。

1.3、在python中导入pytorch成功截图

二、数据集

本次实验使用的是coco数据集中的植物病虫害数据集。分为训练文件traindata和测试文件testdata.，

traindata有9种分类，每一种分类有100张图片。

testdata有9中分类，每一种分类有10张图片。

在我下一篇博客中将数据集开源。

下面是我的数据集截图：

三、代码复现

3.1、导入第三方库

import torchfrom torch.utils.data import datat, dataloaderimport numpy as npimport matplotlibimport osimport cv2from pil import imageimport torchvision.transforms as transformsimport torch.optim as optimfrom torch.autograd import variableimport torch.nn as nnimport torch.nn.functional as ffrom test.cnn import netimport jsonfrom test.train_data import mydatat,pad_image

3.2、cnn代码

# 构建神经网络class net(nn.module):#定义网络模块    def __init__(lf):        super(net, lf).__init__()        # 卷积，该图片有3层，6个特征，长宽均为5*5的像素点，每隔1步跳一下        lf.conv1 = nn.conv2d(3, 6, 5)        #//(conv1): conv2d(3, 6, kernel_河南一本院校size=(5, 5), stride=(1, 1))        lf.pool = nn.maxpool2d(2, 2)#最大池化        #//(pool): maxpool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=fal)        lf.conv2 = nn.conv2d(6, 16, 5)#卷积        #//(conv2): conv2d(6, 16, kernel_size=(5, 5), stride=(1, 1))        lf.fc1 = nn.linear(16*77*77, 120)#全连接层，图片的维度为16，        #(fc1): linear(in日记格式怎么写_features=94864, out_features=120, bias=true)        lf.fc2 = nn.linear(120, 84)#全连接层,输入120个特征输出84个特征        lf.fc3 = nn.linear(84, 7)#全连接层，输入84个特征输出7个特征    def forward(lf, x):        print("x.shape1: ", x.shape)        x = lf.pool(f.relu(lf.conv1(x)))        print("x.shape2: ", x.shape)        x = lf.pool(f.relu(lf.conv2(x)))        print("x.shape3: ", x.shape)        x = x.view(-1, 16*77*77)        print("x.shape4: ", x.shape)        x = f.relu(lf.fc1(x))        print("x.shape5: ", x.shape)        x = f.relu(lf.fc2(x)) 年终考核表       print("x.shape6: ", x.shape)        x = lf.fc3(x)        print("x.shape7: ", x.shape)        return x

3.3、测试代码

img_path = "testdata/test_data/1/apple2 (1).jpg" #使用相对路径image = image.open(img_path).convert('rgb')image_pad = pad_image(image, (320, 320))input = transform(image_pad).to(device).unsqueeze(0)output = f.softmax(net(input), 1)_, predicted = torch.max(output, 1)score = float(output[0][predicted]*100)print(class_map[predicted], " ", str(score)+" %")plt.imshow(image_pad) # 显示图片

四、训练结果

4.1、loss损失函数

4.2、 acc

4.3、单张图片识别准确率

四、小结

这次搭建的网络是基于深度学习框架lenet，并自己做了一些修改完成。最终的训练的结果loss接近0，acc接近100%。但是一般的识别率不会达到这么高，该模型可能会过拟合。可采取剪枝等操作减小过拟合。

到此这篇关于pytorch深度学习之实现病虫害图像分类的文章就介绍到这了,更多相关pytorch图像分类内容请搜索www.887551.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持www.887551.com！