在Windows中的Tensorflow上加载经过训练的ResNet 50网络

Question

我最近开始研究深度学习。我需要运行在Windows CPU上使用Tensorflow加载的经过预训练的ResNet50网络。我找不到在任何地方在线进行操作的结构化方法。有人可以帮我这个忙。

Answer 1

您必须遵循tensorflow网站上的官方“如何进行转移学习”教程。 https://www.tensorflow.org/hub/tutorials/image_retraining

您需要更改的内容是在retrain.py行1302上，它将从tensorflow-hub加载预训练的模型：

default=(
          'https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1'),

对此：

default=(
          'https://tfhub.dev/google/imagenet/resnet_v1_50/feature_vector/1'),

或者您可以在此网站上查看其他经过预先训练的模型列表： https://tfhub.dev/s?module-type=image-feature-vector