求助
木南310803 发布于2021-08 浏览:1626 回复:1
0
收藏
快速回复
最后编辑于2021-08

自己制作了一个coco类型的数据集,类别和coco标准数据集不同,有新的类别,想用预训练模型做迁移学习,求大佬们指条明路怎么实现哇

收藏
点赞
0
个赞
共1条回复 最后由YY应觉月光寒回复于2021-08
#2YY应觉月光寒回复于2021-08

是分类任务吧,调整模型最后一层节点数和你的数据集类别数一致后,做fine tuning就行吧。

想要玩得深入一些的话,还可以考虑冻结网络中用于基础特征提取的前面几层。

0
TOP
切换版块