基于resnet的动物图像分类系统（python期末大作业）PyQt+Flask+HTML5+PyTorch+源代码+文档说明

# pythonexp ## 基于resnet的动物图像分类系统（python期末大作业）PyQt+Flask+HTML5+PyTorch ## requirements ### python=3.7 torch flask tensorboard request pyqt5 # （一）需求和规格说明 **问题描述：**本系统是一款用于动物分类的图像识别系统，基于ResNet残差神经网络。用户可以通过输入关键词，利用爬虫从百度图片中获取相应的动物图像数据集。提供数据集的自动分割和标签加载功能。通过简单的GUI界面，用户可以上传一张图片，系统将自动调用模型进行预测，并在屏幕上显示该图片内动物的种类。 **编程任务：** (1) 编写爬虫脚本，使得可以读取用户输入关键词去爬取下载相应图片数据集。 (2) 实现数据集的自动分割测试集和加载标签。 (3) 建立神经网络模型，编写训练脚本。 (4) 编写GUI界面，使得用户可以传入一张图片，程序需自动调用模型，预测图片类别并打印输出，分别使用PyQt和flask+h5编写本地客户端和网页端。 # （二）设计 ## 1．设计思想 本题主要考察python爬虫知识和深度学习框架的熟练度，作为大作业的题目，该题可以很好体现python所具有的其他语言没有的软件生态。需要用到的爬虫操作需要用户输入一个关键词比如“猫”程序去百度图库搜索并下载对应图片。通过分割脚本整理数据集，然后训练模型保存到特定为止，编写一个简单的GUI，分别使用PyQt和flask+h5编写本地客户端和网页端，使得用户可以选择输入一个图片，并且由模型预测输出类别。 本系统采用了综合应用爬虫、深度学习和图形用户界面（GUI）技术，旨在为用户提供一个简便而有效的动物图像分类系统。整体设计思想围绕以下几个关键点展开： \- **数据采集与处理：** \- 用户输入关键词后，系统利用爬虫技术从百度图片中获取相应的动物图像数据集。这一步骤涉及到异步网络请求、图片下载、数据清洗等操作，确保获得高质量的训练数据。 \- **深度学习模型：** \- 采用ResNet残差神经网络作为图像分类的核心模型。ResNet结构的特点在于可以很好的解决梯度消失或者爆炸、退化问题，提高了模型的准确性。 \- **模型训练脚本：** \- 系统提供了模型的训练脚本，用户可以选择自行训练模型以满足特定需求。模型训练过程中，采用了学习率调整和参数初始化等技术，以提高训练效果。 \- **用户交互界面：** \- 使用PyQt5库创建简单的图形用户界面，使用户能够轻松上传图片、调用模型进行预测并查看结果。GUI设计注重用户友好性和易用性，以提升系统的使用体验。 \- 为了方便快捷，还使用了flask+HTML5部署了网页端，使得随时随地从公网访问服务成为了可能 ## 2. 设计表示代码结构： 文件模块结构： |-- spider爬虫模块 ​ |-- spider.py # 用于爬取图片数据集的爬虫脚本 ​ \# 该脚本定义了一个用于从百度图片搜索下载图片数据集的爬虫。用户可以输入关键词，脚本将在百度图片中搜索相关图片并下载保存。 |-- 数据处理模块 ​ |-- calc_mean.py # 用于计算图片色素均值，用于归一化的脚本 ​ \# 该脚本定义了一个用于计算图片色素均值的工具。色素均值对于训练深度学习模型时的数据预处理非常重要，用于归一化图像。 ​ |-- generate_dataset.py # 生成训练集和测试集的脚本 ​ \# 该脚本用于将下载好的图片数据集分割为训练集和测试集，并生成对应的标签文件（train.txt 和 test.txt）。 |-- 神经网络深度学习模块 ​ |-- train.py # ResNet模型的定义以及训练神经网络模型的脚本 ​ \# 该文件定义了ResNet模型的结构，包括残差块（Residual Block）等。这是系统中用于图像分类任务的主要神经网络模型。该脚本还定义了一个用于训练ResNet模型的脚本。它加载数据集，训练神经网络，并保存训练好的模型权重。 ​ |-- utils.py # 数据加载和预处理的工具函数 ​ \# 该文件包含了一些用于加载和预处理数据的工具函数，如LoadData类，用于加载图片数据集，并进行数据预处理。 |-- pyqt可视化模块 ​ |-- window.py # 主窗口的GUI设计 ​ \# 该文件定义了系统的主窗口GUI，用户可以通过该窗口上传一张图片，系统将自动调用模型进行预测，并在界面上显示预测结果。 ​ |-- predict.py # 预测图片类别的脚本 ​ \# 该脚本定义了一个用于预测单张图片类别的函数。它加载训练好的ResNet模型，读取输入图片，进行预测，并返回预测结果。 |-- images # 用与存放pyqt的背景图像，和用户上传的临时图片 ​ |-- logo.png # 系统图标 ​ \# 该文件是系统的图标，用于在窗口标题栏和任务栏中显示。 ​ |-- horse.png # 预测用的示例图片 ​ \# 该文件是一个用于系统测试的示例图片，用户可以通过GUI上传其他图片进行测试。 |-- templates 文件夹 ​ |-- index.html # 用于部署flask的h5页面代码 |-- output 文件夹 ​ |-- resnet18_e_best.pth # 训练好的ResNet模型的权重 ​ \# 该文件是经过训练得到的ResNet模型的权重文件，用于在GUI中加载预训练的模型。 |-- dataset 文件夹 ​ |-- class1 # 动物类别class ​ |-- image1.jpg ​ |-- image2.jpg ​ \# 该目录包含一个用于训练的数据集。用户可以将自己的数据集放在类似的目录结构下，通过增强数据集提高模型性能。 ​ |-- class2 ​ |-- image3.jpg ​ |-- image4.jpg |-- test.txt # 测试集文件 ​ \# 该文件是测试集的标签文件，记录了测试集中每张图片的路径和对应的类别。 |-- train.txt # 训练集文件 ​ \# 该文件是训练集的标签文件，记录了训练集中每张图片的路径和对应的类别。 ## 3. 核心算法 训练神经网络模型 训练神经网络的核心算法代码在 train.py 文件中。首先，通过 LoadData 类加载训练集和测试集。然后，使用 resnet18 作为基础模型，修改最后一层的全连接层，使其适应问题的输出类别。训练过程中，采用交叉熵损失函数和 Adam 优化器优化损失函数，使用tensorboard来查看训练过程。为了更好地调整学习率，对不同层次的参数采用不同的学习率，对最后一层。训练过程中，记录训练和验证的损失，并保存模型参数。 ResNet模型定义 ResNet 模型的定义在也在train.py 文件中。继承了 resnet18 预训练模型的参数，对其中的最后一层全连接层进行了修改，以适应特定问题的输出类别。这个自定义的 ResNet 模型被用于训练和预测。 数据加载和预处理 数据加载和预处理的工具函数在 utils.py 文件中。LoadData 类负责加载训练集和测试集，该类继承自pytorch框架的dataset类，拥有父类的方法，可以同时进行图像的预处理，包括大小调整、随机翻转、标准化等操作。 图像预测 图像预测的核心算法在 predict.py 文件中。通过加载训练好的 ResNet 模型参数，读取用户输入的图片，进行与训练时相同的预处理操作，然后使用模型进行预测。最终输出预测的动物类别。 # （三）用户手册 本系统是一款基于 ResNet 的动物分类系统，旨在帮助用户通过上传图片实现对动物种类