首页 > 解决方案 > 在哪里可以找到 tensorflow hub 上的模型接受了哪些预处理?

问题描述

我发现 keras.applications 和 tensorflow hub 中可用的一些模型使用了不同的预处理。例如,我读到 mobilenetv2 模型期望图像从 tensorflow hub 缩放 [0,1] 并在 keras.applications 中缩放 [-1, 1]。

现在 keras.applications 模型都带有预处理功能,或者至少带有如何预处理的信息。我可以在 tensorflow hub 上查找相同的信息吗?

不幸的是,我找不到任何东西。

标签: tensorflowkerastransfer-learningtensorflow-hub

解决方案


TF Hub 依赖模型的发布者来记录模型支持的预处理(如果有的话)。也鼓励接口的共同使用(https://www.tensorflow.org/hub/common_signatures),尽管一些发布者可能会遵循他们自己的约定。


推荐阅读