文件名称:Pytorch实现AlexNet花分类完整数据集+代码
文件大小:425.68MB
文件格式:RAR
更新时间:2022-09-21 19:06:50
深度学习 pytorch alexnet模型 花分类
本项目已包含花分类数据集和训练好的AlexNet.pth模型,可直接进行花分类预测,代码共分5部分: 1. 数据预处理:划分数据集 2. 加载自定义数据集 3. 建立Alexnet模型 4. 模型训练 5. 模型的评估和使用训练好的模型进行图片分类 本文代码简单易懂,有较具体的注释,只需具备基础的python知识,便可以顺序通读。