CT影像诊断新冠肺炎
最后更新 2020/06/15 14:34
阅读 9445
PyTorch EfficientNet
yunshao555
12
获得赞0
发布的文章3
答辩的项目Batch大小为16,循环次数为30次,通过在线上环境完成训练,模型最优精度评分为94.12。
最后更新 2020/06/15 14:34
阅读 9445
PyTorch EfficientNet
数据处理
1、通过线下观察数据,发现图像尺寸普遍在400以上,所以考虑使用的模型图像尺寸输入大小为448
2、使用AutoML中的RandAugment作为图像数据增强
模型选择
1、因为图像尺寸较大,所以考虑使用efficientnet系列作为baseline模型框架我使用的是efficientnet-b2模型作为baseline
训练技巧
1、使用标签平滑损失,用来防止过拟合
class LabelSmoo
请先绑定您的微信账号 点击立即绑定
敬请谅解,如有疑问请联系FlyAI客服