首页 > 解决方案 > tf.keras 中的 MobileNetV2。许多链接但没有有用的信息

问题描述

我想将 mobileNetV2 与 tf.keras 一起使用。

如果在 tensorflow 网站上查看 我发现的keras 应用程序

mobilenet = tf.keras.applications.MobileNetV2()

如果我尝试导入 MobileNetV2 from tensorflow.keras.applications import MobileNetV2

我收到一个错误:

ImportError: cannot import name 'MobileNetV2'

如果我查看Keras2 网站,我确实发现只有少数几个应用程序。mobileNetV2(或 V1)不是其中之一。但是V1模型可以加载使用。

如果我点击 tensorflow.keras 网站上的链接,它会将我带到经典的keras 网站,在我看来是 Keras1 而不是 keras2,我错了吗?还说明了显然没有实现的 MobileNetV2。所以我猜链接是错误的。

这一切都让我感到困惑。可能,这都是由于切换到 tf.keras 所致,还是把事情搞混了?

为了更具体地表达我的问题:是否有预定义的、可用的 MobileNetV2 应用程序与 tf.keras 或者我必须手动实现它?

谢谢

编辑:TF 1.10 版。

标签: tensorflowkerastf.kerasmobilenet

解决方案


您正在使用链接作为 MobileNetV2 的参考,但该链接已记录在 tensorflow 版本 1.13 中。而且您使用的是 tensorflow 1.10 版。在此您只能找到 MobileNet 而不是 MobileNetV2。

对于 tensorflow 1.10 版本,您可以像这样导入,

from tensorflow.keras.applications.mobilenet import MobileNet

或者

model = tf.keras.applications.MobileNet()

如果您想检查模型包含在什么中,您可以使用适当的 tensorflow 版本tf.keras.applications检查 github repo 。

如果您想使用 MobileNetV2,请升级您的 tensorflow 版本,您可以按照文档中的说明使用。


推荐阅读