ViT工程文件夹为主要实验模型。训练后工程文件打包有3G多无法上传邮件，需要先训练 训练：运行train.py等待模型训练 预测：运行predict.py 目录： --flops.py --my_dataset.py --predict.py //预测脚本 --train.py //训练脚本 --utils.py
--vit_model.py //主要模型文件 --flower_photos //训练样本 --daisy --dandelion --roses --sunflowers --tulips --weights //训练生成的模型结果 --test1.jpg //测试图片 --jx_vit_base_patch16_224_in21k-e5005f0a.pth //预训练模型权重文件

ResNet文件夹为对比实验代码，代码较少，jupyter打开，自动拆分训练集测试集，仅作对比实验。 训练样本图片过多未上传，请联系我发送网盘链接