tinymind 书法文字识别

版权声明: https://blog.csdn.net/qq_35606924/article/details/80181588

CSDNtinymind 书法文字识别的比赛,

比赛很简单,就是一个多分类问题,直接一把梭renext50,finetune一下就行,稍微用点技巧ok!
下面是比赛链接
https://www.tinymind.cn/competitions/41?from=home1#overview

可以参考我用的代码,

用mxnet训练,调了两个下午,
大约0.985上下
更新了一下,0.989+

直接clone代码
https://github.com/610265158/tinymind_competition.git

下载一个resnext50的预训练模型,

bash ./downmodel.sh

将数据集解压到当前目录,包括一个train,和test1还有一个提交样例

划分数据集

python get_list.py --ratio 0.9

可以训练了

bash ./train.sh

大约几个epoch就可以收敛

然后输出提交结果
这里我们用第四个epoch的模型来预测结果

python predict.py --epoch 6

默认输出一个resnext50_result.csv 可以作为结果提交,

里面也有ensemble的代码, 也可以玩耍一下,不多述

猜你喜欢

转载自blog.csdn.net/qq_35606924/article/details/80181588