0

我正在研究使用 tensorflow 训练对象检测网络，并查看了 TF2 模型动物园。我注意到那里的模型明显少于目录 /models/research/models/ 中的模型，包括为 jetson xavier 开发的带有 ssdlite 的 mobiledet。

澄清一下，自述文件说有一个带有 ssdlite 的 mobildet gpu，并且提供了在 COCO 上训练的模型和检查点，但我在 repo 的任何地方都找不到它们

应该如何使用这些模型？

根据 mobilenetv3 论文，我已经有一个经过定制训练的用于图像分类的 mobilenetv3，我希望看到一种将网络变成对象检测网络的方法。如果这不简单，从头开始训练一个网络也可以，我只需要知道从哪里开始

4

1 回答 1

0

如果您打算使用对象检测 API，则不能使用现有模型。您必须在此处为 v2 和此处为 v1从模型列表中进行选择

文档维护得很好，TensorFlow 团队在此处很好地解释了对自定义数据进行训练、验证或运行推理（测试）的步骤。该链接适用于 TensorFlow 版本 v2。但是，如果您希望使用 v1，则该过程非常相似，并且有许多博客/视频解释了如何进行

于 2021-03-24T14:30:09.330 回答