2020中国华录杯·数据湖算法大赛—定向算法赛(吸烟打电话检测)第二名开源
作者:fire15
联系邮箱:
队伍名称:一个人要像一支队伍
成绩:
初赛A榜:99.84 / 第二名 (两周周冠军)
初赛B榜:84.09 / 第七名
复赛A榜:99.15 / 第五名
复赛B榜:93.13 / 第二名
决赛答辩:第二名
比赛地址:
一、算法简介
数据处理时,观察很多手机拍摄高清图片分辨率很大,因为我机器不大好,IO比较耗时,为了加速训练,先预先等比缩放保存好;
然后合并初赛的数据,进行一个清洗、去重;
最后通过定制化的CutMix进行数据预增强,先生成好部分图片。
主要训练方式基础模型是EfficientNet-B4,使用了对抗学习的预训练模型,使用了Ranger优化器,使用了SGDR的学习率衰减策略,然后通过监测验证集的mAP分数来选择最好的模型,使用early stop停止训练。在训练时还
2021-06-21 15:27:25
45KB
Python
1