[train hyper-parameters: Namespace(batch_size=16, num_classes=6, lr=0.001, lrf=0.01)]
[epoch: 1]
train loss:0.0451 test accuracy:0.8966
[epoch: 2]
train loss:0.0149 test accuracy:0.8807
[epoch: 3]
train loss:0.0075 test accuracy:0.9225
[epoch: 4]
train loss:0.0051 test accuracy:0.9304
[epoch: 5]
train loss:0.0026 test accuracy:0.9344
[epoch: 6]
train loss:0.0017 test accuracy:0.9423
[epoch: 7]
train loss:0.0011 test accuracy:0.9423
[epoch: 8]
train loss:0.0009 test accuracy:0.9463
[epoch: 9]
train loss:0.0007 test accuracy:0.9463
[epoch: 10]
train loss:0.0006 test accuracy:0.9443
没有合适的资源?快使用搜索试试~ 我知道了~
Swin-Transformer 图像分类网络实战项目:6种生活垃圾图像分类数据集(迁移学习)
共2000个文件
jpg:1992个
py:4个
txt:2个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 194 浏览量
2024-03-18
15:16:03
上传
评论 1
收藏 734.08MB ZIP 举报
温馨提示
基于svit网络对6种生活垃圾图像分类数据集的迁移学习项目,包含代码、数据集和训练好的权重文件,经过测试,可以直接运行 【网络】swin-transformer,参数量为8676,8849左右 【数据集】6种生活垃圾图像分类数据集(硬纸板、玻璃、金属、纸张等等6个类别) 【训练train.py】 1、训练过程中,会对数据集进行随机裁剪、翻转等数据增广等等。 2、网络初始化会自动载入官方在imagenet上的预训练权重进行迁移学习。 3、训练脚本会自动生成数据集类别的 json 文件,并且通过json文件自动设定网络的输出维度,不需要自己定义输出的channel。train.py 训练完成会生成训练集的loss曲线、学习率衰减曲线、测试集的accuracy曲线以及训练日志,保存在run_results文件内。 【预测predict.py】只需要将待预测的图像放在inference文件夹下,代码会自动将该文件下所有的图像进行预测,并在原图像左上角写入最大的前三个类别和概率,无需更改任何代码 【本项目训练了10个epoch,准确率为0.95】
资源推荐
资源详情
资源评论
收起资源包目录
Swin-Transformer 图像分类网络实战项目:6种生活垃圾图像分类数据集(迁移学习) (2000个子文件)
paper41.jpg 57KB
paper488.jpg 55KB
paper379.jpg 54KB
paper82.jpg 53KB
paper472.jpg 49KB
metal319.jpg 49KB
paper214.jpg 48KB
paper114.jpg 47KB
paper100.jpg 45KB
paper47.jpg 45KB
paper496.jpg 45KB
paper121.jpg 44KB
paper42.jpg 44KB
paper109.jpg 44KB
paper3_result.jpg 44KB
paper581.jpg 44KB
paper351.jpg 43KB
paper326.jpg 43KB
paper192.jpg 43KB
paper439.jpg 43KB
paper34.jpg 43KB
paper492.jpg 43KB
paper380.jpg 42KB
paper360.jpg 42KB
paper163.jpg 42KB
paper510.jpg 42KB
paper490.jpg 42KB
paper464.jpg 42KB
paper227.jpg 42KB
metal94.jpg 42KB
paper254.jpg 42KB
paper108.jpg 41KB
paper209.jpg 41KB
paper103.jpg 41KB
paper553.jpg 41KB
paper319.jpg 41KB
metal196.jpg 41KB
paper335.jpg 40KB
paper290.jpg 40KB
paper558.jpg 40KB
metal52.jpg 40KB
metal46.jpg 40KB
paper269.jpg 39KB
paper297.jpg 39KB
paper20.jpg 39KB
paper83.jpg 39KB
paper364.jpg 39KB
paper73.jpg 39KB
paper266.jpg 39KB
paper217.jpg 38KB
paper26.jpg 38KB
paper542.jpg 38KB
paper418.jpg 37KB
paper358.jpg 37KB
paper575.jpg 37KB
paper566.jpg 37KB
metal77.jpg 37KB
paper86.jpg 36KB
paper28.jpg 36KB
paper544.jpg 36KB
metal258.jpg 36KB
paper308.jpg 36KB
paper513.jpg 36KB
trash91.jpg 35KB
paper307.jpg 35KB
paper537.jpg 35KB
paper392.jpg 35KB
paper286.jpg 35KB
paper236.jpg 35KB
metal388.jpg 35KB
paper90.jpg 35KB
paper365.jpg 34KB
paper198.jpg 34KB
paper565.jpg 34KB
paper372.jpg 34KB
paper265.jpg 34KB
cardboard266.jpg 34KB
paper415.jpg 34KB
metal383.jpg 34KB
metal251.jpg 34KB
paper457.jpg 34KB
paper14.jpg 34KB
paper74.jpg 34KB
paper92.jpg 34KB
metal118.jpg 34KB
paper362.jpg 33KB
paper512.jpg 33KB
paper245.jpg 33KB
paper451.jpg 33KB
paper206.jpg 33KB
metal16.jpg 33KB
paper563.jpg 33KB
paper312.jpg 33KB
paper285.jpg 33KB
paper338.jpg 33KB
paper487.jpg 33KB
paper578.jpg 33KB
paper194.jpg 32KB
paper231.jpg 32KB
cardboard13.jpg 32KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1239
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- python代码案例详解-旅行商问题的多种求解算法.zip
- springcloud.rar
- 一键清理maven项目依赖下载失败生成的.lastUpdated文件, 以及解决无法连接远程仓库导致本地依赖找不到的批处理脚本
- C#+WPF开发IOC框架实战架构师课程
- 《JAVA WEB项目开发案例精粹》中的代码案例源文件(超全).rar
- 背景music,用作业网轮地址
- comment_6e56cd5d6e140046ef1cd248910ea5ca.png
- 基于卷积神经网络的自动驾驶系统的设计与实现
- MySQL的执行原理-2.md
- 基于 FCN 网络对无人机俯拍下的生活场景语义分割(12类别)【包含数据集、完整代码、训练好的结果、权重文件等等】
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功