模型 = hub.load(module_handle) 抛出错误

问题描述

import tensorflow_hub as hub

module_handle = "https://tfhub.dev/google/faster_rcnn/openimages_v4/inception_resnet_v2/1"
model = hub.load(module_handle)

以上代码片段在 pycharm 环境中抛出如下错误。知道为什么会抛出错误吗?

OSError: SavedModel file does not exist at: OSError: SavedModel file does not exist at: C:\Users\sethu\AppData\Local\Temp\tfhub_modules\6e850c920451d5243d1fb87a3242c087535b9183/{pmodel.ptxt_model.

>

解决方法

你的代码对我有用,所以 URL 看起来不错。请注意 TF Hub caches models locally。您是否检查过此目录和文件是否存在?

C:\Users\sethu\AppData\Local\Temp\tfhub_modules\6e850c920451d5243d1fb87a3242c087535b9183/{saved_model.pbtxt|saved_model.pb

每个 this post 尝试删除目录并再次运行您的代码。