VGG 模型测试仓库

1. 初始化了VGG模型 在花分类数据集上操作
2. 准备向VGG模型中加入SE注意力机制,观察模型性能是否有提高

训练时需要修改的路径: train.py: main 函数中的 data_root,将其改为自己数据集的路径

花分类数据集的组织形式
data/:
train/:
daisy/:
dandelion/:
roses/:
sunflowers/:
tulips/:
val/:
daisy/:
dandelion/:
roses/:
sunflowers/:
tulips/:
上述的 / 表示文件夹,在最终的文件夹中存放着属于各类花的图片