本项目基于 ResNet-50 深度神经网络模型进行图像分类,使用 PyTorch 框架。项目支持图像预处理、数据增强、迁移学习、模型训练与验证、模型保存、推理等功能,适用于图像分类任务。以番茄图像为例,进行多类别的图像分类任务,可以识别不同类型的番茄病害。 项目特点: 使用预训练的 ResNet-50 模型,进行迁移学习。 数据增强...