北京垃圾分类识别
最后更新 2020/06/19 17:54
阅读 13265
EfficientNet PyTorch 图像分类
yunshao555
12
获得赞0
发布的文章3
答辩的项目Batch大小为32,循环次数为20次,通过在线上环境完成训练,模型最优精度评分为85.97。
最后更新 2020/06/19 17:54
阅读 13265
EfficientNet PyTorch 图像分类
1、数据处理
①图片输入网络尺寸,这个可以参考选择的baseline在ImageNet上使用的图片尺寸,或者通过观察数据集图片分布来决定,因为我使用的baseline是efficientnet-b3a,所以选择的图片输入尺寸为320 x 320'efficientnet_b3a': _cfg(input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0)
②数据增强,出于方便,训练阶段直接适用RandAugment作为数据增强操作,测试阶段适用保持长
请先绑定您的微信账号 点击立即绑定
敬请谅解,如有疑问请联系FlyAI客服