kaggle竞赛: Planet: Understanding
2019-06-09 本文已影响0人
深度学习模型优化
使用fastai来看看如何用卫星图像识别亚马逊的植物(有意思)。
这是一个多标签的问题,即在一副图像中有多个标签,我们使用fastai的ImageDataBunch的from_csv来获取数据,并进行数据增强,然后调用现有的模型进行fine-tune,最后使用一些优化技巧和多轮训练,来得到模型,并根据我们的模型来识别结果,提高submission。
上面是基本流程。
1 一些技巧(留存,用来以后使用)
在设置cnn_learner的时候,其metrics参数由两部分组成
acc_02 = partial(accuracy_thresh, thresh=0.2)
f_score = partial(fbeta, thresh=0.2)
learn = cnn_learner(data, arch, metrics=[acc_02, f_score], model_dir='/tmp/models')