2024-05-06 09:05:38
719
551
训练过程中，记录训练和验证的损失，并保存模型参数。通过分割脚本整理数据集，然后训练模型保存到特定为止，编写一个简单的GUI，分别使用PyQt和flask h5编写本地客户端和网页端，使得用户可以选择输入一个图片，并且由模型预测输出类别。通过加载训练好的 ResNet 模型参数，读取用户输入的图片，进行与训练时相同的预处理操作，然后使用模型进行预测。最终输出预测的动物类别。上传图片： 点击“上传图片”按钮，系统会上传图片至服务器，系统后端使用训练好的模型对上传的图片进行动物分类，结果将在界面中显示。_图像分类设计系统界面