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 回答