如何使用tensorflow/models/research/object_detection/models下的模型

how to use the models under tensorflow/models/research/object_detection/models

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

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

应该如何使用这些模型?

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

如果您打算使用对象检测API,则不能使用您现有的模型。您必须从模型列表 here for v2 and here 中选择 v1

文档维护得很好,TensorFlow 团队here 很好地解释了自定义数据的训练或验证步骤或运行 推理(测试)步骤。 link 适用于 TensorFlow 版本 v2。但是,如果您想使用 v1,过程非常相似,并且有很多 blogs/videos 解释如何去做