
图像分类
文章平均质量分 84
图像分类系列
咚咚锵咚咚锵
俯首甘为孺子牛
展开
-
使用grad-cam对swin transformer的特征进行可视化
使用grad-cam对swin-transformer模型的特征图进行可视化原创 2022-09-20 10:00:28 · 10945 阅读 · 16 评论 -
Swin-Transformer图像分类
本机为Ubuntu系统,为了训练自己的数据集,在原代码的基础上做了一点小调整:原代码中每个epoch保存一个模型,调整为只保存表现最佳的模型和最后一个epoch的模型;原代码训练的ImageNet数据集,数据类别比较多,输出了两个评估指标:Top-1 Acc和Top-5 Acc,但我自己数据集只有3个类别,调整为输出Top-1 Acc和Top-2 Acc;原代码没有推理脚本,简单补充了一个原创 2022-09-13 16:36:52 · 4642 阅读 · 18 评论