最后编辑于2021-08
自己制作了一个coco类型的数据集,类别和coco标准数据集不同,有新的类别,想用预训练模型做迁移学习,求大佬们指条明路怎么实现哇
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
是分类任务吧,调整模型最后一层节点数和你的数据集类别数一致后,做fine tuning就行吧。
想要玩得深入一些的话,还可以考虑冻结网络中用于基础特征提取的前面几层。