微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何在迁移学习中向tensorflow.js模型添加新类?

如何解决如何在迁移学习中向tensorflow.js模型添加新类?

一个可以训练对象的预训练网络。如何在不丢失原始类的先前知识的情况下向该模型添加更多类?

我已经阅读了另一篇文章,该文章建议创建一个新模型,除了最后两层外,其他层均具有“基本模型”层。 (为什么要最后2个?)。并将新模型的最后2层权重设置为基本模型的最后2层权重。链接到帖子:如何为深度学习模型添加新类别

weights_training = base_model.layers[-2].get_weights()
new_model.layers[-2].set_weights(weights_training)

请注意,上面的代码在tensorflow中而不是tfjs中。

我能在tensorflow.js中做同样的事情吗?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。