迁移学习和fine-tuning的区别

举个例子,假设今天老板给你一个新的数据集,让你做一下图片分类,这个数据集是关于Flowers的。问题是,数据集中flower的类别很少,数据集中的数据也不多,你发现从零训练开始训练CNN的效果很差,很容易过拟合。怎么办呢,于是你想到了使用Transfer Learning,用别人已经训练好的Imagenet的模型来做。做的方法有很多: 把Alexnet里卷积层最后一层输出的特征拿出来,然后直接用S
相关文章
相关标签/搜索