图片分类模型训练及Web端可视化预测（上）——模型搭建、训练及预测

图像分类模型搭建、训练和预测

任务详情：基于Pytorch框架搭建常见的卷积神经网络，实现对花分类数据集进行训练，并将训练准确率最高的权重保存来进行预测。

1. 花分类数据集介绍

1.1 下载数据集

文件夹详情：

dataset 
├── flower_photos     
		└── daisy
		└── dandelion
		└── roses
		└── sunflowers
		└── tulips

下载链接：
链接: https://pan.baidu.com/s/1dk5uhkH5u3hjF5bzlYyFWw 提取码: t7ix

1.2 数据集划分代码

划分数据集代码 ——> train, val
split.py

im