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

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

如何解决如何使用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 和 here for v1

的模型列表中进行选择

文档得到很好的维护,TensorFlow 团队很好地解释了在自定义数据上训练或验证或运行推理(测试)的步骤here。该链接适用于 TensorFlow 版本 v2。但是,如果您想使用 v1,该过程非常相似,并且有许多博客/视频解释了如何操作

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