使用keras.models的load_models加载facenet-keras模型时卡住

vecaoik1  于 2023-05-18  发布在  其他
关注(0)|答案(2)|浏览(193)

我尝试使用python3加载facenet-keras模型,但我的代码使用keras加载facenet模型,并将tensorflow作为后端。我的代码如下:

from keras.models import load_model
print("Loading facenet model...")
model = load_model('models/facenet_keras_dup.h5') #code stucks at this line
print ("facenet model loaded")

我的python版本如下:
python 3.5.2 keras 2.3.1 tensorflow 1.14
运行于Intel i3第6代3.7 Ghz CPU和Intel HD Graphics 530

zpgglvta

zpgglvta1#

我想你会得到一个类似这样的错误:

数值错误:封送数据错误

此错误可能是由于不同版本的python或keras或tensorflow。我认为我们得到的错误,因为版本通过facenet模块保存可能是不同的,从我们的版本

我是通过创建虚拟环境,安装以下版本的软件包解决的:
python:3.7.1,Keras:2.3.1,Tensorflow:2.2.0

注意:这是一个尝试和错误的解决方案和版本是一个给我的解决方案。不是标准版本所需的facenet模型

6l7fqoea

6l7fqoea2#

**如果你有pip,那么首先使用命令安装这些:

pip install keras-facenet

pip install tensorflow

pip install keras-resnet

导入这些库

import tensorflow as tf
from keras_facenet import FaceNet

导入FaceNet模型

model = tf.keras.applications.ResNet50(weights='imagenet')

这个应该能用

相关问题