前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >图像之场景分类 实验研究

图像之场景分类 实验研究

原创
作者头像
XianxinMao
修改2021-09-01 18:04:25
4420
修改2021-09-01 18:04:25
举报
文章被收录于专栏:深度学习框架深度学习框架

首先把数据集的情况放上来,如下图所示:

然后再看一下要用到的一些组件,如下图所示:

最后一起看下模型的训练流程:

最后一起看下训练的结果日志,如下图所示,可以看到 testAcc = 93.86%,此处用的是双卡 2080Ti,比单卡来说,训练的 batch_size 更大,所以准确率略微提升:

还有一些地方可以总结下:

模型在训练过程当中的指标有 loss 和 accuracy,从长期来看 loss 呈下降趋势,accuracy 呈上升趋势

为了加快训练速度,可以在训练过程中不计算 train_acc,取 val_acc 最大处,保存的 checkpoint

earlyStop 特别适用于 train_acc 和 val_acc 的 gap 随训练不断扩大,如果 val_acc 在 5 个 epoch 未发生下降,则停止训练

LR Scheduler,如果 train_loss 在 3 个 epoch 未发生下降,则学习率发生衰减

训练速度和过拟合是两大瓶颈

LR Scheduler 和 train_loss、train_acc 挂钩,test、val 和泛化性相关

代码地址: https://github.com/MaoXianXin/PycharmProjects

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档