tensorflow - 使用 Keras 预训练的视网膜网络模型。SavedModel 文件不存在于：“路径”/“文件”.h5/{saved_model.pbtxt|saved_model.pb}

Question

我正在尝试通过运行使用 keras 加载预训练的视网膜网络模型：

# import keras
import keras

# import keras_retinanet
from keras_retinanet import models
from keras_retinanet.utils.image import read_image_bgr, preprocess_image, resize_image
from keras_retinanet.utils.visualization import draw_box, draw_caption
from keras_retinanet.utils.colors import label_color


# set tf backend to allow memory to grow, instead of claiming everything
import tensorflow as tf

def get_session():
    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    return tf.Session(config=config)
model_path = os.path.join('sample_data/snapshots', sorted(os.listdir('sample_data/snapshots'), reverse=True)[0])
print(model_path)


# load retinanet model
model = models.load_model(model_path, backbone_name='resnet50')

model = models.convert_model(model)

两个代码我都面临以下错误：

OSError: SavedModel file does not exist at: sample_data/snapshots/training_5000(640_480).h5/{saved_model.pbtxt|saved_model.pb}

原因可能是一些新版本的 Keras 或 tensorflow，所以我将列出我当前使用的版本。

keras.__version__
2.4.3

tf.__version__
2.4.1

注意：我正在尝试在我的 Colab 中运行此代码。

标签： tensorflowkerasdeep-learningcomputer-visionretinanet