10

我已经从http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz在 Inception v3 的 ImageNet 上下载了一个预训练模型（在遵循其中一个教程时找到了这个链接代码实验室）。

这个 zip 文件包含 .pb 文件，我可以直接将其作为 TensorFlow 中的 GraphDef 导入。

我想知道其他架构是否也有类似的 .pb 文件，例如 ResNet、VGG16、MobileNet 等。如果有，你能提供这些的链接吗？

提前致谢。

亲切的问候，

阿杰

4

2 回答 2

9

您可以在这里找到许多预训练模型：https ://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

并且对应的代码在这里： https ://github.com/tensorflow/models/tree/master/research/slim/nets

于 2018-05-07T12:56:45.893 回答
2

以下是下载预训练模型的其他来源：

虽然略有不同，但 Keras 提供了一个访问预训练模型的接口：

于 2021-09-14T17:03:38.493 回答