如何基于coco-ssd进行迁移学习

Question

老师，您好，我想基于 coco-ssd 进行迁移学习，咋们课程示例给的是LayerModel , tfjs-models/coco-ssd 示例给出的是GraphModel, 看了官方文档，这两个模型差异还挺大的, 应该怎样进行迁移学习？

Answer 1

lewis 2021-02-24 14:37:16

tfjs 支持转换 graph model，看一下文档哈

• 提问者 广州_小彭 #1

  看过了文档, tf.GraphModel 模型的参数都是固定的，没有layers, 不能进行截断, 再进行迁移学习

• lewis 回复 提问者 广州_小彭 #2

  目标检测建议用python训练，训练完后将graph model转为tfjs model进行预测

• 提问者 广州_小彭 回复 lewis #3

  谢谢老师，明白了