上传者: hhhhhhhhhhwwwwwwwwww
|
上传时间: 2022-02-24 09:11:40
|
文件大小: 930.94MB
|
文件类型: ZIP
本例提取了植物幼苗数据集中的部分数据做数据集,数据集共有12种类别,演示如何使用timm版本的Swin Transformer图像分类模型实现分类任务已经对验证集得分的统计,本文实现了多个GPU并行训练。
通过本文你和学到:
1、如何从timm调用模型、loss和Mixup?
2、如何制作ImageNet数据集?
3、如何使用Cutout数据增强?
4、如何使用Mixup数据增强。
5、如何实现多个GPU训练和验证。
6、如何使用余弦退火调整学习率?
7、如何使用classification_report实现对模型的评价。
8、预测的两种写法。