首页 PaddleCV 帖子详情
在图片分类实践,新增了图片类别,能否固定原来参数不变,训练新的参数
收藏
快速回复
PaddleCV 其他图像分类 1649 3
在图片分类实践,新增了图片类别,能否固定原来参数不变,训练新的参数
收藏
快速回复
PaddleCV 其他图像分类 1649 3

这个想法我以前也想过,相信很多小伙伴也这么想过。实际上,这个想法理论上可行,但效果恐怕不会太好。如果只是固定全连接层,而前面的卷积层参数发生了变化,那么这些全连接层的作用也无法保证与开始一样。而如果保持整个网络的参数都不变,只训练全连接层的新增两个类别,也比较难训练处理想的结果。

0
收藏
回复
全部评论(3)
时间顺序
DeepGeGe
#2 回复于2021-12

一般来说,如果实际使用中确实需要原有的 100 个类别依然很准确,那么可以将新类别的数据加入到原有训练集中,再用预训练模型进行 finetune。如果只需要 100 个类中的几个类别,那么可以把这部分的数据挑出来,和新增数据混合再 finetune。

0
回复
stivenssss
#3 回复于2021-12

你可以固定个别层的参数,比如resnet50的,你开放卷积层中倒数几层的,然后固定其他的,再把分类输出的个数改为新增后总的类别数,就可以了

0
回复
退后是人生
#4 回复于2021-12

可以试试

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户