## Vision Transformer(ViT)
# 代码会自动载入预训练权重
1. 准备好自定义数据集,按照下面格式摆放
data-train---后面跟文件夹,每个文件夹里面存放相同的图像,几个文件夹代表几个分类
data-test
2. 在 `train.py` 脚本中更改下面超参数
# num_classes 必须根据分类的个数进行更改 !!!其他的看着改
parser.add_argument('--num-classes', type=int, default=5)
parser.add_argument('--epochs', type=int, default=100)
parser.add_argument('--batch-size', type=int, default=16)
parser.add_argument('--lr', type=float, default=0.001)
parser.add_argument('--lrf', type=float, default=0.01)
parser.add_argument('--freeze-layers', type=bool, default=True) # 是否冻结权重
3.`predict.py`脚本会自动预测 inference 文件夹下的所以图片
# 下面的三个都要更改,num_classes 是自己分类的个数
# mean,std 是训练图片的均值和方差,训练的时候会生成,copy在这里就行('./run_results/train_log_results.txt' 文件里也有)
num_classes = 5
mean = [0.467,0.424,0.304]
std = [0.301,0.271,0.296]
main(mean=mean,std=std,num_classes=num_classes)
没有合适的资源?快使用搜索试试~ 我知道了~
Vision Transformer 网络对7种小麦叶片病害分类,包含训练权重和数据集、迁移学习
共2000个文件
jpg:1988个
py:4个
xml:4个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 188 浏览量
2024-05-01
18:30:31
上传
评论
收藏 831.46MB 7Z 举报
温馨提示
1.网络使用的是Transformer 网络中的 ViT 模型 2.数据集是7分类的小麦叶片病害(水泡,褐色斑点) 3.网络训练的过程中,会对数据集的训练集图像求取灰度值的均值和方差,预处理处理包括随机裁剪、中心裁剪等等。网络初始化会自动载入ViT 官方的预训练权重(transformer网络收敛很慢,不经过预训练效果会很差,因此本篇代码自动载入) 4.train.py 会自动生成数据集类别的 json 文件,所以不需要自己定义。train.py 训练完成会生成训练集的loss、测试集的loss,训练集的accuracy、测试集的accuracy,并生成对应的曲线保存至run_results文件内。训练完成会计算训练集和测试集的混淆矩阵,计算召回率、特异度啥的 5.预测代码是predict.py,只需要将待预测的图像放在inference文件夹下,代码会自动将该文件下所有的图像进行预测,并在原图像左上角写入最大的前三个类别和概率 【如果想要训练自己的数据集,参考README文件】 【本项目测试了30个epoch,准确率为0.88,网络未收敛,加大轮次可以增加精度】
资源推荐
资源详情
资源评论
收起资源包目录
Vision Transformer 网络对7种小麦叶片病害分类,包含训练权重和数据集、迁移学习 (2000个子文件)
.4359ab1e6493cc12248bd29dcd9281e6_result.jpg 27KB
Hispa-697-_jpg.rf.09a4faa3bef550dcc83a738050dc7d0b.jpg 26KB
shape-433-_jpg.rf.56390ea731c0a8acc603ffc817337dd8.jpg 26KB
Hispa-603-_jpg.rf.f02aa3d946cade31f93ee68e05f31d4c.jpg 26KB
shape-541-_jpg.rf.d4202dc01c96c7d85d814df40c7bb4ed.jpg 26KB
Hispa-638-_jpg.rf.94143ca3a0f8ef8e51e0bbb68d71ea2b.jpg 26KB
Hispa-112-_jpg.rf.9262afd3d1c821da68b36b34a4827642.jpg 25KB
shape-457-_jpg.rf.153ed8805e450b8ef74f4a601a9cbb12.jpg 25KB
shape-510-_jpg.rf.c713eda63c6f41cd62ef775ce9fb5920.jpg 25KB
shape-149-_jpg.rf.0cfbc7cbd76d30ea1dc4afde3156336f.jpg 25KB
shape-75-_jpg.rf.1f4b474d880a95a0862c404792aa02a0.jpg 25KB
shape-511-_jpg.rf.2b7d7a587d666529c3b6f15d0636e8f5.jpg 25KB
shape-363-_jpg.rf.5cce9d53c043dfd95a575afb5c1caaf4.jpg 25KB
shape-403-_jpg.rf.9a3862694ce8ba0ab0addc35bf9a3e78.jpg 25KB
shape-268-_jpg.rf.aeaa8287a5de2e33cfdd2351314d424e.jpg 25KB
shape-275-_jpg.rf.86230d27c5c8c5f680b5e83a7d6b1f86.jpg 25KB
shape-520-_jpg.rf.0eec0710992976d26ffdfb86070424fd.jpg 25KB
IMG_20190419_124756_jpg.rf.8ca808b4cfe348fdf6e07c465294bfb9.jpg 25KB
Hispa-672-_jpg.rf.d84b5ddd97192d2b186cba490f1a72df.jpg 25KB
Hispa-731-_jpg.rf.96eb3a50422bcbb8932456d2e46c8d1b.jpg 25KB
shape-361-_jpg.rf.c1c211b0a1365c13815dc3dee240e60c.jpg 25KB
shape-49-_jpg.rf.25a18b98c4b4a082771205bffd5e271c.jpg 25KB
shape-115-_jpg.rf.a80572a1d16bf0b962bd88030b3c354c.jpg 25KB
shape-235-_jpg.rf.ad1f59cca094ff1a00d03b3a78478dd9.jpg 25KB
shape-12-_jpg.rf.456a178233f67fdc23ee1e8c0d9f113c.jpg 25KB
shape-80-_jpg.rf.202d72d62b79c434d23fadd059269154.jpg 25KB
shape-295-_jpg.rf.3739e87159b1b018cbf2ce0bc079f9e9.jpg 25KB
Hispa-555-_jpg.rf.12a55943a302fce7091824d85e96fcc4.jpg 25KB
IMG_20190419_134839_jpg.rf.9fd2ce58223d7dfc250e899ff5d34a3f.jpg 25KB
shape-146-_jpg.rf.779968b1f622a1949f6c0ab9409a4ee5.jpg 25KB
Hispa-468-_jpg.rf.01011a6fe9114afce7747fb4a87674bf.jpg 25KB
shape-141-_jpg.rf.4d3eaec232fb5219e21990e0cfcc66f9.jpg 25KB
Hispa-600-_jpg.rf.0e735e7ba8032acbc8b279d676bece57.jpg 25KB
shape-185-_jpg.rf.b14a3741743b6738c04c1fa040a61743.jpg 25KB
shape-105-_jpg.rf.eea13137c5ece0241085c3f2bc0d1171.jpg 25KB
shape-162-_jpg.rf.7e3adff2f3730da009d982d752086d08.jpg 25KB
shape-408-_jpg.rf.bc7c319ae02bdf367dc5946f03b2bf21.jpg 25KB
Hispa-480-_jpg.rf.9996a214e931b652c1d68752b24846d7.jpg 25KB
shape-411-_jpg.rf.32fef80e1e3d8eb6f1fae97650a6425b.jpg 25KB
Hispa-161-_jpg.rf.cbcd3ab347733432d74965f3ec074daf.jpg 25KB
IMG_20190419_114834_jpg.rf.f3e2e2cf8044d6d9cdc70cce3a4493bf.jpg 25KB
shape-564-_jpg.rf.8a2204b34115beacb4cbacabd1c97001.jpg 25KB
shape-446-_jpg.rf.d7ff29ed98ea44daa8a95e5020f7fa6a.jpg 25KB
IMG_20190419_095930_jpg.rf.d6d14c0e090b5b5a55db84398914b759.jpg 25KB
shape-530-_jpg.rf.5e6b4fbc1145ff3dee540925f3b4d64b.jpg 25KB
Hispa-113-_jpg.rf.b1f24eb26260ddc8c68805402a100d9f.jpg 25KB
Hispa-481-_jpg.rf.2e390dbceaf2df87b7e15e5cdaeb4854.jpg 25KB
shape-163-_jpg.rf.fb8af698ded8cf200310bff6525ffd49.jpg 25KB
shape-241-_jpg.rf.20719e4e073f827a08bb1f091dcc37ab.jpg 25KB
shape-233-_jpg.rf.70494470e118be7484439084e5e6a0e4.jpg 25KB
IMG_20190419_110457_jpg.rf.2743436aaffbf4da0e1d7f1f7918e6cc.jpg 25KB
shape-417-_jpg.rf.a661535b716478ff1b7898fbc33b44c1.jpg 25KB
shape-54-_jpg.rf.79170e248eabb036dee3c160f57d6f44.jpg 25KB
Hispa-468-_jpg.rf.ef4f8631879610f2be1c0fc0b34b2ebb.jpg 25KB
Hispa-618-_jpg.rf.b9a0d25ae9216d016231ff17f8b7be25.jpg 25KB
shape-92-_jpg.rf.54ab67299a25e45e60094a59bf67d38f.jpg 25KB
shape-38-_jpg.rf.0c3e6b60e2973c1afcbb89a58c568869.jpg 25KB
Hispa-286-_jpg.rf.fe5a51ab4b3be27d31df586127e5647c.jpg 25KB
shape-428-_jpg.rf.9f9c279c27d8a03024268b1371d88e17.jpg 25KB
shape-159-_jpg.rf.001295f0b412072da21e9ebd142d7fbd.jpg 25KB
IMG_20190419_140059_jpg.rf.7b08c5e06f28412c1cf76e675a868e36.jpg 25KB
shape-63-_jpg.rf.a2b4cd872e0aac679a8717be8142c0e8.jpg 25KB
shape-526-_jpg.rf.4a8387319651797b226ebb10a2ebbc57.jpg 25KB
shape-506-_jpg.rf.b889a3dfe494c3a71bc940ab97265c32.jpg 24KB
Hispa-212-_jpg.rf.906b95100ee93ecdb610bc5df574b818.jpg 24KB
shape-222-_jpg.rf.2c234f0e0330866f47f0e7b362153d8d.jpg 24KB
shape-167-_jpg.rf.4826d81f74b5de2896336c58ebe6efc8.jpg 24KB
shape-408-_jpg.rf.132f4b22ef966ef32554ac72008d340c.jpg 24KB
shape-514-_jpg.rf.3bd017015064c61251d80479c78e6199.jpg 24KB
IMG_20190419_144604_jpg.rf.19b22c0d8aff629d7bb735fb58e35009.jpg 24KB
shape-90-_jpg.rf.1761ec0749198c944e7105dd9a42b2d2.jpg 24KB
shape-340-_jpg.rf.7b3c4a8745f2facf4392c224a2756399.jpg 24KB
shape-390-_jpg.rf.6a4cc2e00e443f67a30c38a2057b6fa7.jpg 24KB
IMG_20190419_144654_jpg.rf.127cc52a04009a4015f0bb06a1c61afb.jpg 24KB
shape-138-_jpg.rf.eed22ae3d9d7e7ce2c3070e35fe09502.jpg 24KB
shape-509-_jpg.rf.e2caba8cd5aa04acce4d4ab0f73a1c18.jpg 24KB
shape-28-_jpg.rf.afdd7451cd3518935a433c89f224e7ad.jpg 24KB
shape-56-_jpg.rf.9f729d4d9c2a12df174f365b7c86802d.jpg 24KB
shape-42-_jpg.rf.482a05f5d9bc52c25736c3fa82df7f59.jpg 24KB
shape-83-_jpg.rf.021e7ac1001ec35df44bf4b55725c743.jpg 24KB
Hispa-685-_jpg.rf.3c43c718ebbb9cc3df77826c78a4c8e1.jpg 24KB
Hispa-678-_jpg.rf.3029cf095b17135595cc370a765c41c2.jpg 24KB
shape-403-_jpg.rf.26cc43d64d3c2d5edda7f9896544e11a.jpg 24KB
Hispa-579-_jpg.rf.2018cf8be3f6cf70b922a6cb3ca37e82.jpg 24KB
shape-302-_jpg.rf.7e68e79dd5ff60086c5e3ea3d8e00940.jpg 24KB
Hispa-212-_jpg.rf.fce2fcaa40decb86b2d2adbd6cb29fb7.jpg 24KB
Hispa-459-_jpg.rf.fccf7207dc6a9c7615e6f2e25c89795f.jpg 24KB
shape-24-_jpg.rf.8c4a2b03ab8c403d758a5159d3e2877c.jpg 24KB
shape-519-_jpg.rf.d926b7b4ec1a5f6ea499c2a137193e44.jpg 24KB
shape-465-_jpg.rf.0fe8589d953b67f81ded66c14de7716b.jpg 24KB
shape-336-_jpg.rf.81869fab83efd991e19791d47244c6ab.jpg 24KB
shape-34-_jpg.rf.c0415f3f781f2d373de62112ff0ac41e.jpg 24KB
IMG_20190419_115003_jpg.rf.796c75d2587af38f1c458ca0554cae15.jpg 24KB
shape-463-_jpg.rf.fc03101cfd673c84bfd2c844b58a38ec.jpg 24KB
shape-59-_jpg.rf.4289e57d3870db57409e72d37f313a08.jpg 24KB
Hispa-648-_jpg.rf.9be41326b4c26fb141d4c596d948d591.jpg 24KB
shape-184-_jpg.rf.a4a183a27db6513c47717073ff68eae7.jpg 24KB
Hispa-691-_jpg.rf.2e80a4f637bcccfc7a34dbea5487ddb9.jpg 24KB
Hispa-607-_jpg.rf.4c14c188fa89271eddcbcc081023b818.jpg 24KB
shape-11-_jpg.rf.9460e43847261cda422740058b34623e.jpg 24KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1220
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功